python - 混淆矩阵不支持多标签指示符

Question

不支持多标签指示器是我在尝试运行时收到的错误消息:

confusion_matrix(y_test, 预测)

y_test 是一个 DataFrame，其形状为:

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

预测是一个numpy数组:

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

我已经搜索了一些错误消息，但还没有真正找到我可以应用的内容。有什么提示吗？

Answer 1

否，您输入 confusion_matrix必须是预测列表，而不是 OHE(一种热门编码)。在您的 y_test 和 y_pred 上调用 argmax，您应该会得到您所期望的结果。

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

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