from sklearn.metrics import roc_curve, auc画ROC曲线

本文探讨了ROC曲线的基本原理,它用于评估分类模型的性能,特别是正负样本的区分能力。在二分类任务中,可以使用sklearn库的roc_curve函数绘制ROC曲线,并计算AUC值。对于多分类任务,将得到多条曲线并可能计算平均AUC。作者强调实践中遇到的问题、参考官方文档和多角度学习的重要性。此外,还提到了在深度学习中如何从模型输出构建ROC曲线,特别是通过概率排序和设定阈值的方法。
摘要由CSDN通过智能技术生成

一、明白ROC的原理

ROC曲线针对的是 正负样本,画出预测曲线。

网上给出的概念解读有很多,我查看的这篇 ROC and AUC, Clearly Explained! - YouTube

我说说自己对ROC的理解,ROC曲线是用来判断 预测效果 的。深度学习对猫狗分类做ROC曲线时,使用from sklearn.metrics import roc_curve, auc是可以画出两条ROC曲线的。一条是猫,另一条是狗的ROC,最后取做平均。

二、from sklearn.metrics import roc_curve, auc

roc_curve(y_true,y_score,*,pos_label=None,sample_weight=None,drop_intermediate=True):                           

参数详解:参考Python下使用sklearn绘制ROC曲线(超详细)_hesongzefairy的博客-CSDN博客。需要注意的时,在标签是二进制时,pos_label默认是1。

在n分类任务时,画出的ROC曲线是n条。我们需要对n条曲线做平均。

三、关于roc曲线的其他画法

在深度学习二分类中,test的output返回的是二维张量。通过torch.argmax(outputs)得出最大概率的下标以及概率值。我们对这个概率值进行排序,设定阈值来画出ROC曲线。

 四、感受

1、问题需要坚持、实践

2、参考 官方文档,博客大多具有一定的适应范围

3、请教别人,多方面请教

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值