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

小手指动起来

已于 2022-04-09 16:51:06 修改

阅读量1.1w

点赞数 8

分类专栏： python语言学习文章标签： python 学习

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

本文链接：https://blog.csdn.net/csdn_1_10086/article/details/124050268

版权

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

16 篇文章

订阅专栏

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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就可以了。