AI 漫天,我们坚持手做论坛。

新加坡 10-30 16:10 · 吉隆坡 16:10 · 雅加达 15:10 · 曼谷 15:10 · 马尼拉 16:10 · 胡志明 15:10 · 北京 16:10 · 东京 17:10 · 伦敦 08:10 · 纽约 04:10

计算二分类ROC AUC时,传入预测分数并核对它属于哪一类

#模型评估#ROC#AUC#分类

1
新加坡 · 中文 · r/人工智能 · Amy · 声望 381 · 10-30 14:14 新 · 帖子

Amy 原创整理|资料核验:2026-10-30

scikit-learn的roc_auc_score文档说明,二分类的y_score可以是概率估计,也可以是未经过阈值切分的decision_function分数。使用predict_proba时,要对应类别标签较大的那一类,通常是estimator.classes_[1]对应的列。

教学解释:把概率先按0.5压成0和1,会丢掉原本的分数排序信息;即使程序仍能算出一个值,也不等于对连续模型分数完成了同样的评估。

Amy建议:计算前检查真实标签、classes_顺序、分数方向及样本对齐。先用一个可人工检查的小例子确认,再处理完整验证集。多分类输入还有概率列顺序和参数要求,不能直接照搬二分类取一列的写法。

官方来源:
https://scikit-learn.org/stable/modules/generated/sklearn.metrics.roc_auc_score.html

回复

还没有回复。

广告由 Google AdSense 投放,标为广告,与帖子分开。请勿点击自己的广告,也不要请别人点。行情讨论不是荐股。本地服务帖写地点和价格,线下交易注意安全。

关于 · 隐私 · 联系 · 版规