tabensemb.model.AutoGluon._train_single_model

method

AutoGluon._train_single_model(model, model_name, epoch, X_train, y_train, X_val, y_val, verbose, warm_start, in_bayes_opt, **kwargs)

Training the model (initialized in _new_model()).

Parameters:

model:

The model returned by _new_model().

model_name:

The name of the model.

epoch:

Total epochs to train the model.

X_train:

The training data from _train_data_preprocess().

y_train:

The training target from _train_data_preprocess().

X_val:

The validation data from _train_data_preprocess().

y_val:

The validation target from _train_data_preprocess().

verbose:

Verbosity.

warm_start:

Finetune models based on previous trained models.

in_bayes_opt:

Whether is in a Bayesian optimization loop.

**kwargs:

Parameters to train the model returned by _get_params(). It contains all arguments in _initial_values().