scikit-learn return value of LogisticRegression.predict_proba scikit-learn return value of LogisticRegression.predict_proba python python

scikit-learn return value of LogisticRegression.predict_proba


4.65761066e-03 + 9.95342389e-01 = 19.75851270e-01 + 2.41487300e-02 = 19.99983374e-01 + 1.66258341e-05 = 1

The first column is the probability that the entry has the -1 label and the second column is the probability that the entry has the +1 label. Note that classes are ordered as they are in self.classes_.

If you would like to get the predicted probabilities for the positive label only, you can use logistic_model.predict_proba(data)[:,1]. This will yield you the [9.95342389e-01, 2.41487300e-02, 1.66258341e-05] result.