tabensemb.model.TorchModel._predict_all

method

TorchModel._predict_all(verbose: bool = True, test_data_only: bool = False) → Dict[str, Dict]

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.