- reproducible results on a single machine via the
seeds
parameter. - proper CUDA support. Automatic detection if you have a CUDA-enabled machine.
- enable logistic output/binary classification. Occurs automatically when
torch.nn.BCELoss()
function passed to thecriterion
parameter.
- fixed
init_test_size
of model selection functions working on the full dataset including NAs in the target variable
- make model instantiation and model selection robust to variables with no data in them
- feature contribution weighting for data availabilty in interval predict functions
- got rid of hardcoded references to date column in interval predict code
- ability to generate uncertainty intervals via the
model.interval_predict()
function - ability to generate uncertainty intervals on synthetic vintages via the
model.ragged_interval_predict()
function
initial_ordering
parameter tovariable_selection()
andselect_model()
functions. In recursive feature addition (RFA) variable selection, can obtain initial variable order either via their feature contribution in a full model, or from univariate model performances. Former (default) is about 2x faster.
- ability to obtain feature contributions to the model via
model.feature_contribution()
function - automatic variable selection given a set of hyperparameters via
variable_selection()
function inLSTM.model_selection
- automatic hyperparameter tuning given a set of variables via
hyperparameter_tuning()
function inLSTM.model_selection
- automatic variable and hyperparameter tuning via
select_model()
function inLSTM.model_selection
- hide printing of
Training model n
whenquiet=True