基于sklearn.metrics.roc_auc_score的几种多分类AUC计算方式


 

以A、B、C三类为例,步骤如下:

1、        测试集共m个样本,分别计算每个样本与对比集的cos距离(对比集由A、B、C三类构成),根据结果构建概率矩阵P[m*3],再基于one-hot构建标签矩阵L[m*3];

2、        P的每一列即各样本属于该列对应类别的概率,则可以计算每个类别的ROC曲线和auc值;

3、        计算整体ROC曲线和auc值

(1)   方法一:macro,对n条ROC曲线取平均,即可得到最终的ROC曲线

分析:该方法不考虑类别不均衡的影响;

(2)   方法二:micro,参考下面

 

 

 

计算总的TP rate和FP rate,然后计算ROC曲线和auc值。

(3) 方法三:weighted,通过每个类别的TP数所占比例进行加权平均;

备注:目前sklearn.metrics.roc_auc_score(仅支持macro 和 weighted)


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM