不,您输入的内容confusion_matrix
必须是预测列表,而不是OHE(一种热编码)。调用argmax
您的y_test
和y_pred
,您应该得到期望的结果。
confusion_matrix(
y_test.values.argmax(axis=1), predictions.argmax(axis=1))
array([[1, 0],
[0, 2]])
混淆矩阵不支持Multilabel-indicator
不,您输入的内容confusion_matrix
必须是预测列表,而不是OHE(一种热编码)。调用argmax
您的y_test
和y_pred
,您应该得到期望的结果。
confusion_matrix(
y_test.values.argmax(axis=1), predictions.argmax(axis=1))
array([[1, 0],
[0, 2]])