Python代码解析:绘制二分类混淆矩阵热力图
这段代码使用了seaborn库和matplotlib库来绘制一个二分类混淆矩阵的热力图。\n\n首先,通过导入seaborn库和sklearn.metrics库中的confusion_matrix函数,以及matplotlib库中的pyplot模块,准备工作完成。\n\n然后,使用y_true和y_pred两个数组来表示真实标签和预测标签。这里的y_true表示真实标签,y_pred表示预测标签。\n\n接下来,使用confusion_matrix函数来计算混淆矩阵C2。其中,labels参数指定了标签的类别,这里有三个类别0、1、2。\n\n然后,使用print语句打印出混淆矩阵C2的值。\n\n接着,使用seaborn库中的heatmap函数,将混淆矩阵C2绘制成热力图,annot参数设置为True表示在热力图上显示数值,ax参数指定了绘图的坐标轴。\n\n接下来,使用matplotlib库中的pyplot模块来设置图表的字体为楷体(KaiTi),并设置图表的标题为'confusion matrix\n 二分类混淆矩阵 李涵博',x轴和y轴的标签分别为'predict'和'true'。\n\n最后,使用show函数显示图表。
原文地址: https://www.cveoy.top/t/topic/pA26 著作权归作者所有。请勿转载和采集!