了解logreg.predict_proba(X_test)[:, 1])

时间:2019-05-10 05:15:31

标签: python scikit-learn

我正在使用此函数来计算预测的概率 logreg.predict_proba(X_test)[:,1])

我对目标变量为(0(良好),1(不良))的数据集进行了逻辑回归。

一件事,我无法理解[:,1]的含义是什么? 并且如果我更改为logreg.predict_proba(X_test)[:,0]),我的auc_roc_score的值也会更改,因此我应该使用

logreg.predict_proba(X_test)[:,1])

logreg.predict_proba(X_test)[:,0])

0 个答案:

没有答案