tabensemb.model.AbstractModel._predict_model_on_partition#

method

AbstractModel._predict_model_on_partition(model_name: str, partition: str) ndarray[source]#

Get predictions of a model on the selected partition.

Parameters:
model_name

The selected model.

partition

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