我正在使用此函数来计算预测的概率
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])