校准器(Calibrated Learner)
用概率校准和决策阈值优化来包装另一个学习者。
输入
- 数据:输入数据集
- 预处理器:预处理方法
- 基学习器:要校准的学习器
输出
- 学习器:校准的学习算法
- 模型:使用经过校准的学习器训练的模型
功能
该学习器生成一个模型,该模型可校分类概率的分布并优化决策阈值。 该小部件仅适用于二元分类任务。
界面
- 它将在其他小部件中显示的名称。 默认名称由学习器,校准和优化参数组成。
- 概率校准:
- 决策阈值优化:
- 优化分类准确性
- 优化 F1 score
- 无阈值优化
- 发送报告
- 勾选 “自动应用” 以自动传送对其他小部件的更改,并在连接学习数据后立即训练分类器。 或者,在配置后按 “应用”。
示例
我们使用 titanic 数据集,因为小部件需要二元类值(在这种情况下,它们是“生存”和“未生存”的)。
我们将使用逻辑回归(Logistic Regression)作为基学习器,我们将使用默认设置进行校准,即通过分布值的 sigmoid 优化和通过优化准确率进行校准。
将结果与未经校准的 逻辑回归(Logistic Regression) 模型进行比较,我们可以看到校准后的模型表现更好。
参考文献
- Why Calibrators? Part 1 of the Series on Probability Calibration
- Calibration
- 1.16. Probability calibration
反馈问题
文档有问题? 或者有其他意见和建议? 请在本文档的 Github 仓库直接反馈
点我反馈进入反馈页面不知道如何反馈, 请点击这里