如何创建混淆矩阵?

import numpy as np
import matplotlib.pyplot as plt
from sklearn.metrics import confusion_matrix

# 模型预测结果和实际标签
y_pred = [10, 22, 5, 5, 14, 10, 36, 22, 14, ...]
y_true = [10, 22, 5, 3, 14, 10, 14, 22, 14, ...]

# 创建混淆矩阵
conf_mat = confusion_matrix(y_true, y_pred, labels=range(56))

# 可视化混淆矩阵
plt.figure(figsize=(12, 12))
plt.imshow(conf_mat, interpolation='nearest', cmap=plt.cm.Blues)
plt.title('Confusion Matrix')
plt.colorbar()
tick_marks = np.arange(len(conf_mat))
plt.xticks(tick_marks, tick_marks)
plt.yticks(tick_marks, tick_marks)
plt.xlabel('Predicted Class')
plt.ylabel('True Class')
plt.show()

这段代码使用了sklearn.metrics中的confusion_matrix函数来计算混淆矩阵。你需要将模型的预测结果y_pred和实际标签y_true传递给该函数,并使用labels参数指定分类的数量(这里是56)。然后,使用matplotlib库来可视化混淆矩阵。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值