为什么人们对roc_auc_score()使用predict_proba()的第二列(即predict_proba(X)[:,1])?

时间:2017-10-30 00:11:35

标签: python-3.x machine-learning scikit-learn

我已经多次看到人们使用predict_proba(X)第二列的值(即predict_proba(X)[:,1])来查找二进制分类模型的roc_auc_score(),如下所示:

from sklearn.ensemble import RandomForestClassifier
rfc = RandomForestClassifier()
rfc.fit(X_train, y_train)

ypred = rfc.predict_proba(X_test)[:,1]
roc_auc_score(y_test, ypred)

0 个答案:

没有答案