tabensemb.model.RFE._train

method

RFE._train(verbose: bool = True, model_subset: list | None = None, warm_start=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.