python代码报错：ValueError: multiclass format is not supported（已解决）

多分类问题：ROC曲线误用与修复

原创已于 2022-04-09 16:51:06 修改 · 1.1w 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#python #学习

于 2022-04-09 16:25:57 首次发布

python语言学习专栏收录该内容

16 篇文章

订阅专栏

在深度学习评估中，遇到`ValueError: multiclass format is not supported`错误，发现是由于将多分类标签误用于二分类ROC曲线。解决方法是识别并修正标签，确保为二分类。通过实例代码和原因分析，探讨如何避免此类问题。

ValueError: multiclass format is not supported

找到报错位置：

def compute_auc(pred, label):
    if isinstance(pred, torch.Tensor):
        pred = pred.cpu().detach().numpy().flatten()
    if isinstance(label, torch.Tensor):
        label = label.cpu().detach().numpy().flatten()
    fpr, tpr, thresholds = metrics.roc_curve(label, pred)#报错位置
    return metrics.auc(fpr, tpr)

报错原因：
在这里插入图片描述
设置断点debug看看这两个输入数据的内容和类型

报错的原因就是pos_label=None了

仔细查看label的数值，有0, 1，2，是个多分类，但是roc曲线一般是二分类的，多分类用混淆矩阵来做。我的数据就是二分类，那肯定就是我的label数据有个标签打成了2，改为1就可以了。