混淆矩阵如何用热力图表示,来明显显示出模型的好坏,写出具体的代码
时间: 2024-10-15 20:23:28 浏览: 42
混淆矩阵是一种用于评估分类模型性能的工具,它展示了实际结果和预测结果之间的对比情况。将混淆矩阵转换成热力图,可以帮助直观地看出每个类别间的预测准确度。通常,行代表真实标签,列代表预测标签,颜色越深或值越大表示对应类别的匹配程度高。
以下是使用Python的seaborn库和sklearn库创建混淆矩阵热力图的一个简单示例:
```python
import seaborn as sns
import matplotlib.pyplot as plt
from sklearn.metrics import confusion_matrix
# 假设我们有一个分类模型,y_true是实际标签,y_pred是预测标签
y_true = [0, 1, 0, 1, 0] # 实际标签
y_pred = [0, 0, 1, 1, 0] # 预测标签
# 计算混淆矩阵
conf_mat = confusion_matrix(y_true, y_pred)
# 使用seaborn画热力图
plt.figure(figsize=(8,6))
sns.heatmap(conf_mat, annot=True, cmap='YlGnBu', fmt='g')
plt.xlabel('Predicted Class')
plt.ylabel('True Class')
plt.title('Confusion Matrix Heatmap')
plt.show()
```
在这个例子中,颜色从浅色(低匹配度)到深色(高匹配度),可以清楚地看到模型在各个类别上的表现。如果模型很优秀,主要对角线上的数值会较大,表示大部分预测都是正确的;反之,非对角线上的数值大,说明有较多的误判。
阅读全文