混淆矩阵热力图使用颜色编码来表示混淆矩阵中不同类别之间的关系给出一段代码示例
以下是一个使用颜色编码来表示混淆矩阵的热力图的代码示例:
import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt
# 创建混淆矩阵
confusion_matrix = np.array([[50, 10, 5],
[5, 40, 10],
[10, 5, 45]])
# 使用seaborn库绘制热力图
sns.heatmap(confusion_matrix, annot=True, cmap="Blues", fmt="d")
# 设置图表标题和轴标签
plt.title("Confusion Matrix")
plt.xlabel("Predicted Class")
plt.ylabel("True Class")
# 显示图表
plt.show()
在这个示例中,我们首先创建了一个混淆矩阵confusion_matrix,它是一个3x3的矩阵,表示了3个类别之间的预测结果。
然后,我们使用seaborn库的heatmap函数来绘制热力图。annot=True参数表示在热力图中显示数值,cmap="Blues"参数表示使用蓝色调色板来表示不同数值的颜色,fmt="d"参数表示对数值进行格式化为整数。
接下来,我们设置了图表的标题和轴标签,并最后使用plt.show()函数显示图表。
运行这段代码,你将得到一个表示混淆矩阵的热力图,其中不同类别之间的关系用颜色编码表示
原文地址: https://www.cveoy.top/t/topic/ilzJ 著作权归作者所有。请勿转载和采集!