不,您输入的内容
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]])

不,您输入的内容
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]])