如何才能很好地可视化混淆矩阵？

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

# 准备数据
y_true = np.array([0, 1, 2, 0, 1, 2, 0, 1, 2])
y_pred = np.array([0, 0, 2, 0, 2, 1, 0, 1, 2])

# 计算混淆矩阵
cm = confusion_matrix(y_true, y_pred)

# 可视化混淆矩阵
plt.imshow(cm, cmap=plt.cm.Blues)

# 添加颜色条
plt.colorbar()

# 添加坐标轴标签
plt.xlabel('Predicted Label')
plt.ylabel('True Label')

# 添加类别标签
tick_marks = np.arange(len(np.unique(y_true)))
plt.xticks(tick_marks, np.unique(y_true))
plt.yticks(tick_marks, np.unique(y_true))

# 显示图像
plt.show()