我已经多次看到人们使用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)