混淆矩阵不支持多标签指标 | 混淆矩阵不支持多标签指标

本文介绍了混淆矩阵不支持多标签指标的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！

问题描述

multilabel-indicator is not supported 是我在尝试运行时收到的错误消息:

multilabel-indicator is not supported is the error message I get, when trying to run:

confusion_matrix(y_test, predictions)

y_test 是一个 DataFrame 形状:

Horse | Dog | Cat
1       0     0
0       1     0
0       1     0
...     ...   ...

predictions 是一个 numpy 数组:

[[1, 0, 0],
 [0, 1, 0],
 [0, 1, 0]]

我已经搜索了一些错误消息，但还没有真正找到我可以应用的东西.有什么提示吗?

I've searched a bit for the error message, but haven't really found something I could apply. Any hints?

推荐答案

不，您对 confusion_matrix 必须是预测列表，而不是 OHE(一种热编码).在你的 y_test 和 y_pred 上调用 argmax，你应该得到你期望的结果.

No, your input to confusion_matrix must be a list of predictions, not OHEs (one hot encodings). Call argmax on your y_test and y_pred, and you should get what you expect.

confusion_matrix(
    y_test.values.argmax(axis=1), predictions.argmax(axis=1))

array([[1, 0],
       [0, 2]])

这篇关于混淆矩阵不支持多标签指标的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！