如何将深度学习的混淆矩阵画出来
时间: 2023-10-31 21:04:28 浏览: 131
confusion_混淆矩阵、pytorch、模型_混淆矩阵pytorch_混淆矩阵_
5星 · 资源好评率100%
在深度学习中,可以使用混淆矩阵来评估模型在多类别分类问题上的性能。下面是一个示例代码,展示如何使用混淆矩阵来评估深度学习模型:
```python
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
from sklearn.metrics import confusion_matrix
# 生成实际类别和预测类别
y_true = np.random.randint(low=0, high=3, size=100)
y_pred = np.random.randint(low=0, high=3, size=100)
# 计算混淆矩阵
cm = confusion_matrix(y_true, y_pred)
# 创建标签列表
labels = ['Class 0', 'Class 1', 'Class 2']
# 绘制混淆矩阵热力图
plt.figure(figsize=(8, 6))
sns.heatmap(cm, annot=True, fmt='d', cmap='Blues', xticklabels=labels, yticklabels=labels)
plt.title('Confusion Matrix')
plt.xlabel('Predicted')
plt.ylabel('Actual')
plt.show()
```
上述代码中,我们使用了`seaborn`库来绘制热力图,利用`confusion_matrix`函数计算混淆矩阵,并使用`heatmap`函数进行可视化。在热力图中,每个格子的颜色代表对应类别的分类数量,并使用注释显示具体的数量值。x轴和y轴表示预测类别和实际类别。
阅读全文