ZBLOG

填写代码,画出混淆矩阵

要画出混淆矩阵,可以使用以下代码:

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

# 计算混淆矩阵
cm = confusion_matrix(y_test, lr_yhat)

# 创建标签数组
labels = ['Non-Fraud', 'Fraud']

# 绘制混淆矩阵图像
disp = ConfusionMatrixDisplay(confusion_matrix=cm, display_labels=labels)
disp.plot(cmap='Blues')

# 设置标题和轴标签
plt.title('Confusion Matrix')
plt.xlabel('Predicted Label')
plt.ylabel('True Label')

# 显示颜色条例
plt.colorbar()

# 展示图像
plt.show()

上述代码中,首先通过confusion_matrix()函数计算出混淆矩阵。然后使用ConfusionMatrixDisplay类创建可视化对象,并通过plot()方法绘制混淆矩阵图像。

接着,设置图像的标题和轴标签,并通过colorbar()函数显示颜色条例。

最后,使用show()函数展示图像。

请注意,上述代码中的 y_testlr_yhat 是之前逻辑回归模型预测得到的测试集真实标签和预测结果。确保在运行此代码之前已经进行了逻辑回归模型的训练和预测操作。


内容由零声教学AI助手提供,问题来源于学员提问

本站部分文章来源于网络,版权归原作者所有,如有侵权请联系站长删除。
转载请注明出处:https://golang.0voice.com/?id=14117

分享:
扫描分享到社交APP
上一篇
下一篇
发表列表
游客 游客
此处应有掌声~
评论列表

还没有评论,快来说点什么吧~

联系我们

在线咨询: 点击这里给我发消息

微信号:3007537140

上班时间: 10:30-22:30

关注我们
x

注册

已经有帐号?