本文主要是介绍ValueError: multilabel-indicator is not supported(混淆矩阵不支持multilabel-indicator)，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！
混淆矩阵不支持multilabel-indicator
- 我们要做什么？
- 出错源代码
- 如何更改能够运行
我们要做什么？
我们要通过sklearn包直接获取决策树的混淆矩阵，但是遇到了报错，之前对车辆评估直接调用是没有报错的，但是手写数字识别报错了！
出错源代码
其中：Ytest为划分的测试集数据(为one-hot编码)
clf.predict(Xtest)为预测结果
from sklearn.metrics 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的逆编码
from sklearn.metrics import confusion_matrix
confusion_matrix = confusion_matrix(Ytest.argmax(axis=1),clf.predict(Xtest).argmax(axis=1))
print(confusion_matrix)
这篇关于ValueError: multilabel-indicator is not supported(混淆矩阵不支持multilabel-indicator)的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
