热线电话:13121318867

登录
2020-05-24 阅读量: 944
Python机器学习AUC两种方式

计算AUC的两种方式

1、什么是AUC?

●AUC是ROC曲线下方的面积

●取值一般为0.5-1,越大表明分类性能越好!

4.2 直接使用封装好的API

代码:

from sklearn import metrics

scores = lr.predict_proba(X_test)[:,1]

metrics.roc_auc_score(y_test, scores) # y_test真实标签 scores为预测为1的概率

0.6989812656479323

●有一个坑要注意,roc_auc_score中第一个参数是真实标签值,第二个是预测为1类的概率值!不要弄反了!

●roc_auc_score的官方文档:https://scikit-learn.org/stable/modules/generated/sklearn.metrics.roc_auc_score.html

1.3858
3
关注作者
收藏
评论(0)

发表评论

暂无数据
推荐帖子