Python 混淆矩阵热力图绘制代码详解

这段代码使用 seaborn 库和 matplotlib 库来绘制混淆矩阵的热力图。\n\n首先，导入了 seaborn 库和 sklearn.metrics 库中的 confusion_matrix 函数，以及 matplotlib.pyplot 库。\n\n然后，调用 seaborn 的 set() 函数设置绘图风格。\n\n接下来，创建一个图形和一个坐标轴对象，用于绘制热力图。\n\n然后，定义了真实标签 y_true 和预测标签 y_pred。\n\n使用 confusion_matrix 函数计算混淆矩阵 C2，指定标签为 [0,1,2]。\n\n接着，通过 print 语句打印出混淆矩阵 C2。\n\n使用 sns.heatmap 函数绘制热力图，传入混淆矩阵 C2 和坐标轴对象 ax，并设置 annot=True 来显示数值。\n\n通过 plt.rcParams['font.sans-serif'] = ['KaiTi'] 设置字体为楷体。\n\n使用 plt.title 函数设置标题为 'confusion matrix 李涵博'。\n\n使用 plt.xlabel 函数设置 x 轴标签为 'predict'，使用 plt.ylabel 函数设置 y 轴标签为 'true'。\n\n最后，使用 plt.show() 显示热力图。\n\n重复一次绘制热力图和设置标题、坐标轴标签、显示热力图的步骤。\n\n总结来说，这段代码的主要功能是计算并绘制给定真实标签和预测标签的混淆矩阵的热力图。