tabensemb.model.WideDeep._predict_model

method

WideDeep._predict_model(model_name: str, test_data_only: bool = False) → Dict[str, Tuple]

Get predictions of a model on all partitions.

Parameters:

model_name

The selected model.

test_data_only:

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

Returns:

Its keys are “Training”, “Testing”, and “Validation”. Its values are tuples containing predicted values and

ground truth values.