在Python中怎样绘制校准曲线评估分类模型的置信度概率

梦晨大大_6723

梦晨大大_6723

2026-10-16

617人浏览

原创

calibration_curve返回二分类下各分箱的平均预测概率与真实正例比例,需输入0/1标签和1d概率数组(如y_pred_proba[:, 1]),n_bins建议设10–20;calibratedclassifiercv通过sigmoid或isotonic法校准原始模型概率,不改预测标签只优化概率输出。

在python中怎样绘制校准曲线评估分类模型的置信度概率

用 calibration_curve 获取校准数据点

校准曲线本质是把模型输出的概率按分箱(bin)统计真实正例频率(y_true)和平均预测概率(y_prob),sklearn.calibration.calibration_curve 就是干这个的。它不画图,只返回两个数组:横轴(平均预测概率)和纵轴(实际正例比例)。

常见错误是直接传入多分类的 y_pred_proba 全矩阵——它只接受二分类的 1D 概率数组(比如 y_pred_proba[:, 1])。若用 OneVsRestClassifier 等包装器,也得先提取某类的置信度。

  • 必须确保 y_true 是 0/1 标签,不是 one-hot 或字符串
  • n_bins 默认 5,太粗略;实践中建议设为 10–20,尤其当样本量 > 5000 时
  • 若某 bin 内无样本,calibration_curve 会跳过该 bin,返回的横纵坐标长度可能小于 n_bins

用 CalibratedClassifierCV 对模型做概率校准

原始模型(如 SVC、RandomForestClassifier)输出的概率常不靠谱,CalibratedClassifierCV 可在训练时注入校准逻辑。它支持两种方法:sigmoid(Platt scaling)适合 SVM;isotonic(保序回归)更灵活但易过拟合小数据集。

注意它不改变预测标签,只重映射概率值。校准后模型的 predict_proba 输出才适合喂给 calibration_curve。

  • 对小样本(isotonic,改用 sigmoid 或干脆不校准
  • 使用交叉验证模式(cv='prefit' 除外)时,校准是在每个 fold 上独立做的,最终概率是平均结果,所以无需额外 cross_val_predict
  • CalibratedClassifierCV 的 predict_proba 返回值仍是二维数组,记得取第二列用于二分类校准分析

手动画校准曲线并加参考线

用 matplotlib 把 calibration_curve 返回的点连成线,再叠加 y=x 参考线(理想校准)和直方图(显示各 bin 样本量),信息才完整。

Forge — Python Developer
Forge — Python Developer

Python开发助手 — 自动化脚本、API封装、命令行工具、数据管道、网页爬虫。经过测试的代码,包含错误处理和文档。当你需要...

下载

容易忽略的是:概率分箱不均匀会导致某些 bin 样本极少,曲线抖动大。建议同时画出每个 bin 的样本数(用柱状图或点大小编码),否则可能误判“S 形偏差”其实是稀疏采样噪声。

  • 横纵坐标范围都设为 [0, 1],否则参考线错位
  • 用 plt.scatter 比 plt.plot 更直观,点大小可设为 bin_weights * 50(bin_weights 来自 calibration_curve 的第 3 个返回值)
  • 别忘了加网格:plt.grid(True, alpha=0.3),方便读数值

对比多个模型时避开常见陷阱

想比 LogisticRegression、RandomForest、XGBoost 的校准效果?关键不是看曲线离 y=x 多近,而是看校准前后的 ECE(Expected Calibration Error)变化。ECE 是加权平均绝对误差,权重是各 bin 样本占比。

一个隐蔽坑:不同模型预测概率的分布范围可能不同(比如 RF 常输出偏高或偏低概率),直接画在同一图上会掩盖系统性偏差。应分别计算 ECE,并用箱线图展示多次划分训练/测试集的结果,而非仅单次 split。

  • sklearn.metrics.brier_score_loss 是另一个指标,但它对异常高置信低准确预测更敏感,和 ECE 互补,不要只盯一个
  • 测试集必须和训练集完全隔离——校准步骤(包括 CalibratedClassifierCV 的 cv 过程)不能接触测试标签
  • 如果模型本身不支持 predict_proba(如原生 XGBoost 的 booster='gbtree'),需先用 predict_proba 方法或设置 objective='binary:logistic'

校准曲线看着简单,但 bin 划分方式、样本分布偏斜、模型是否真正支持概率输出——这些细节一错,整条线就失去诊断价值。

Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

python

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1731

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

4444

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1709

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

26037

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3167

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3187

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1183

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

596

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2463

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Uvicorn官方安装与入门手册
Uvicorn官方安装与入门手册

共0课时 | 0人学习

Python uv官方项目管理指南
Python uv官方项目管理指南

共0课时 | 0人学习

Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习