AUC ROC曲线多类分类

Question

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

n_classes= 5

y_test = [0,1,1,2,3,4]   #actual value
pred1 = [0,1,1,1,3,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], tpr[i], _ = roc_curve(np.array(pd.get_dummies(y_test))[:, i], np.array(pd.get_dummies(pred1))[:, i])

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

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

如何解决此错误。