ssl_framework.main.SelfTrainingClassifier.predict

SelfTrainingClassifier.predict(X)

Predict class labels for samples in X.

Parameters:

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

Returns:

y_pred – Predicted class labels for each sample.

Return type:

ndarray of shape (n_samples,)