出现这个状况主要是因为在使用matplotlib绘图时,语句的逻辑顺序有问题。
# 错误的语句顺序
plt.imshow(cm, interpolation='nearest', cmap=cmap)
plt.title(title, fontsize=10)
plt.colorbar()
xlocations = np.array(range(n_classes))
plt.xticks(xlocations, labels_zh_cmn, fontsize=10)
plt.yticks(xlocations, labels_zh_cmn, fontsize=10)
plt.ylabel('True Label', fontsize=10)
plt.xlabel('Pred Label', fontsize=10)
# 正确的语句顺序
plt.title(title, fontsize=10)
xlocations = np.array(range(n_classes))
plt.xticks(xlocations, labels_zh_cmn, fontsize=10)
plt.yticks(xlocations, labels_zh_cmn, fontsize=10)
plt.ylabel('True Label', fontsize=10)
plt.xlabel('Pred Label', fontsize=10)
plt.imshow(cm, interpolation='nearest', cmap=cmap)
plt.colorbar()
错误的语句顺序对应的混淆矩阵图如下:
正确的语句顺序对应的混淆矩阵图如下: