ValueError: multilabel

ValueError: multilabel

混淆矩阵不支持multilabel-indicator

• 我们要做什么？
• 出错源代码
• 如何更改能够运行

我们要做什么？

我们要通过sklearn包直接获取决策树的混淆矩阵，但是遇到了报错，之前对车辆评估直接调用是没有报错的，但是手写数字识别报错了！

出错源代码

其中：Ytest为划分的测试集数据(为one-hot编码)
clf.predict(Xtest)为预测结果

ics import confusion_matrix
confusion_matrix = confusion_matrix(Ytest,clf.predict(Xtest))
print(confusion_matrix)

如何更改能够运行

在Ytest和clf.predict(Xtest)后面添加.argmax(axis=1)就可以运行了
.argmax(axis=1)相当于转化成为一个十进制的数字，相当于从one-hot的逆编码

ics import confusion_matrix
confusion_matrix = confusion_matrix(Ytest.argmax(axis=1),clf.predict(Xtest).argmax(axis=1))
print(confusion_matrix)