好看的混淆矩阵可视化,以小数方式显示

文章展示了如何利用Python的seaborn库和sklearn.metrics的ConfusionMatrixDisplay模块创建混淆矩阵的可视化。通过转换矩阵并用annot和cmap参数定制,可以生成带有注释和自定义颜色的热力图,帮助理解模型预测性能。

在这里插入图片描述
混淆矩阵通常长上面这个样子,通过confusion_matrix(y_pred, y_test) 即可调用。在写论文时一般会用小数形式表示,并且会有很好看的可视化效果,如下:
在这里插入图片描述

只需要简单几行,即可实现

import seaborn as sns
cm_1 = cm/cm.sum(axis=0) #转换成小数
cm_2 = pd.DataFrame(cm_1, index=['A','B','C']
评论 3
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值