tabensemb.model.AutoGluon.predict_proba

method

AutoGluon.predict_proba(*args, **kwargs) → ndarray

Predict probabilities of each class.

Parameters:

args

Positional arguments of predict().

kwargs

Arguments of predict(), except for proba.

Returns:

np.ndarray

For binary tasks, a (n_samples, 1) np.ndarray is returned as the probability of positive. For multiclass tasks, a (n_samples, n_classes) np.ndarray is returned.