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






评论(0)


暂无数据
推荐帖子
0条评论
0条评论
0条评论