如何得到混淆矩阵？错误的原因可能是什么[关闭]

Question

如何得到混淆矩阵？我不明白这个问题

输出：

代码的输出看起来像这样

  (10729, 3) (10729,)
    [[0. 0. 1.]
     [0. 1. 0.]
     [0. 0. 1.]
     [0. 0. 1.]
     [0. 0. 1.]] [2 1 2 2 2]

CLASSES = ['1', '2', '3']
# predict 
pred = model.predict(train_data, batch_size = 32)
pred = np.argmax(train_labels, axis=1)
# label
y_train = np.argmax(train_labels, axis=1)

#train_labels=to_categorical(train_labels-1,3)

print(train_labels.shape, pred.shape)
print(train_labels[:5], pred[:5])