ssl_framework.main.SelfTrainingClassifier.predict_proba#

SelfTrainingClassifier.predict_proba(X)[source]#

Predict class probabilities for samples in X.

Parameters:

X (array-like of shape (n_samples, n_features)) – Samples to predict probabilities for. Can be numpy array or pandas DataFrame.

Returns:

y_proba – Predicted class probabilities for each sample and class.

Return type:

ndarray of shape (n_samples, n_classes)