imblearn.ensemble.RUSBoostClassifier.staged_predict_proba

RUSBoostClassifier.staged_predict_proba(X)[source]

Predict class probabilities for X.

The predicted class probabilities of an input sample is computed as the weighted mean predicted class probabilities of the classifiers in the ensemble.

This generator method yields the ensemble predicted class probabilities after each iteration of boosting and therefore allows monitoring, such as to determine the predicted class probabilities on a test set after each boost.

Parameters
X{array-like, sparse matrix} of shape (n_samples, n_features)

The training input samples. Sparse matrix can be CSC, CSR, COO, DOK, or LIL. COO, DOK, and LIL are converted to CSR.

Yields
pgenerator of ndarray of shape (n_samples,)

The class probabilities of the input samples. The order of outputs is the same of that of the classes_ attribute.