修正micro_auc = roc_auc_score(y_test, y_scores, multi_class='ovo', average='micro')
时间: 2024-02-27 22:58:10 浏览: 136
ROC曲线绘制以及AUC分数计算
感谢您的指正,正确的计算多分类问题中的 AUC 应该是使用 `roc_auc_score` 函数,设置参数 `multi_class='ovr'`,并且不能指定 `average` 参数。`multi_class='ovr'` 表示采用 One-vs-Rest 的策略计算每个类别的 AUC 值,最后取平均值作为评价指标。修正后的计算公式如下:
```
micro_auc = roc_auc_score(y_test, y_scores, multi_class='ovr')
```
其中,`y_test` 是测试集的真实标签,`y_scores` 是分类器预测的概率值或者决策函数输出值。
阅读全文