跳到正文
Towards Data Science· Ankur Tomar·· 3 小时前AI 评分51

Scikit-Learn 在 BANKING77 数据集上校准表现与 Jev 模型对比

Scikit-Learn Matched or Beat Jev's Calibration on BANKING77

AI 导读

作者使用 scikit-learn 实现的逻辑回归模型与 TypeSafe AI 的 Jev 模型在 BANKING77 数据集上进行校准效果对比,发现温度缩放的逻辑回归在7类任务上与Jev的校准效果相当(ECE 0.022 vs 0.014),在77类任务上表现更优(0.031 vs 0.065)。

来源:Towards Data Science · towardsdatascience.com