AUC ROC曲线多类分类

编程问答 2022-06-13

问题描述

这是用于多个类别的ROC AUC曲线计算的代码的一部分。

n_classes= 5

y_test = [0,1,2,3,4]   #actual value
pred1 = [0,4]    #predicted value


fpr = dict()
tpr = dict()
roc_auc = dict()

for i in range(n_classes):
    fpr[i],tpr[i],_ = roc_curve(np.array(pd.get_dummies(y_test))[:,i],np.array(pd.get_dummies(pred1))[:,i])
    roc_auc[i] = auc(fpr[i],tpr[i])

错误：

    fpr[i],i])

IndexError: index 4 is out of bounds for axis 1 with size 4

我了解到，在pred1中仅预测了4个类0,4，在y_test中预测了5个类0,4。因此会出现此错误。

如何解决此错误。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

aucmulticlass-classificationpythonroc