【机器学习要点记录】metrics.roc_auc_score：使用 model.predict() 和 model.predict_proba() 效果差别很大

分类: 文章 • 2022-10-02 13:59:43

以前使用roc_auc_score都是用model.predict()的预测结果，错了那么久。。。。。。

查看官网注释才知道应该用model.predict_proba()的预测结果。

做了个对比试验，分别使用model.predict() 和 model.predict_proba()的预测结果，观察AUC值发现，相比于使用model.predict_proba()的AUC值，使用model.predict()的AUC值偏小。