如何打印图像分类器的混淆矩阵(CIFAR-10)

import numpy as np
import matplotlib.pyplot as plt
from sklearn.metrics import confusion_matrix

# 加载测试集数据和预测结果
test_labels = np.load('test_labels.npy')
predicted_labels = np.load('predicted_labels.npy')

# 计算混淆矩阵
cm = confusion_matrix(test_labels, predicted_labels)

# 可视化混淆矩阵
plt.imshow(cm, interpolation='nearest', cmap=plt.cm.Blues)
plt.title('Confusion Matrix - CIFAR-10')
plt.colorbar()
tick_marks = np.arange(10)
plt.xticks(tick_marks, ['飞机', '汽车', '鸟类', '猫', '鹿', '狗', '青蛙', '马', '船', '卡车'], rotation=45)
plt.yticks(tick_marks, ['飞机', '汽车', '鸟类', '猫', '鹿', '狗', '青蛙', '马', '船', '卡车'])
plt.tight_layout()
plt.ylabel('真实标签')
plt.xlabel('预测标签')
plt.show()