asreview.models.classifiers.NN2LayerClassifier.predict_proba

NN2LayerClassifier.predict_proba(X)

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.