tabensemb.model.AbstractModel._train_single_model

method

AbstractModel._train_single_model(model: Any, model_name: str, epoch: int | None, X_train: Any, y_train: ndarray, X_val: Any, y_val: Any, verbose: bool, warm_start: bool, in_bayes_opt: bool, **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().