tabensemb.model.AutoGluon._train

method

AutoGluon._train(model_subset: List[str] | None = None, dump_trainer: bool = True, verbose: bool = True, warm_start: bool = False, **kwargs)

The basic framework of training models, including processing the dataset, training each model (with/without bayesian hyperparameter optimization), and evaluating them on the dataset.

Parameters:

model_subset:

The names of a subset of all available models (in get_model_names()). Only these models will be trained.

dump_trainer:

Whether to save the trainer after models are trained.

verbose:

Verbosity.

warm_start:

Finetune models based on previous trained models.

**kwargs:

Ignored.