asreview.models.classifiers.base.BaseTrainClassifier.predict_proba

BaseTrainClassifier.predict_proba(X)[source]

Get the inclusion probability for each sample.

Parameters:

X (numpy.ndarray) – Feature matrix to predict.

Returns:

numpy.ndarray – Array with the probabilities for each class, with two columns (class 0, and class 1) and the number of samples rows.