tabensemb.model.PytorchTabular._predict_model_on_partition

method

PytorchTabular._predict_model_on_partition(model_name: str, partition: str) → ndarray

Get predictions of a model on the selected partition.

Parameters:

model_name

The selected model.

partition

“train”, “val”, or “test”.