tabensemb.model.RFE._predict_all#

method

RFE._predict_all(**kwargs)[source]#

Make inferences on training/validation/testing datasets to evaluate the performance of all models.

Parameters:
verbose:

Verbosity.

test_data_only:

Whether to predict only the testing set. If True, the whole dataset will be evaluated.

Returns:
dict

A dict of results. Its keys are names of models, and its values are results from _predict_model() for each model.