在Python中,绘制混淆矩阵通常使用`matplotlib`和`seaborn`库。以下是一个简单的步骤说明和示例代码:
1. 导入必要的库:
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
from sklearn.metrics import confusion_matrix
2. 准备真实标签和预测标签数据:
y_true = [0, 0, 1, 2, 1, 2, 0, 2, 2, 0, 1, 1] 真实标签
y_pred = [1, 0, 1, 2, 1, 0, 0, 2, 2, 0, 1, 1] 预测标签
3. 计算混淆矩阵:
C = confusion_matrix(y_true, y_pred)
4. 使用`seaborn`绘制热力图形式的混淆矩阵:
plt.figure(figsize=(9, 7))
sns.heatmap(C, annot=True, cmap='Blues', fmt='g', xticklabels=np.unique(y_true), yticklabels=np.unique(y_true))
plt.title('Confusion Matrix')
plt.xlabel('Predicted')
plt.ylabel('True')
plt.show()
以上代码将生成一个热力图,其中每个格子的大小和颜色表示该预测类别与实际类别的一致性。`annot=True`表示在格子中显示具体的计数,`fmt='g'`表示数值以整数形式展示。`xticklabels`和`yticklabels`用于设置x轴和y轴的刻度标签,这里使用了数据集中真实的类别标签。
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。
如需转载请保留出处:https://sigusoft.com/bj/145202.html