我只想澄清一下
之间的区别roc_auc_score(y_test,results.predict(X_test))
和
roc_auc_score(y_test,results.predict_proba(X_test)[:,1])
我知道后者会为每次测试观察返回0级的概率,并且在绘制出roc_curve时,应该使用predict_proba()。但是哪种方法可以检查ROC中的二进制分类模型性能?我现在使用前一个。后者是什么意思?
答案 0 :(得分:0)
第二个是正确的 您的预测应该排名,因为ROC AUC只关心排名。