您输入的内容confusion_matrix
必须是整数数组,而不是一种热编码。
matrix = metrics.confusion_matrix(y_test.argmax(axis=1), y_pred.argmax(axis=1))
从Keras多类模型获取混淆矩阵
您输入的内容confusion_matrix
必须是整数数组,而不是一种热编码。
matrix = metrics.confusion_matrix(y_test.argmax(axis=1), y_pred.argmax(axis=1))