如何在 LightGBM 中使用 r2-score 作为损失函数?

数据挖掘 scikit-学习 回归 xgboost 损失函数 lightgbm
2022-02-18 00:59:56

我正在尝试在 LightGBM 中为回归问题实现自定义损失函数。内在指标对我没有多大帮助,因为它们会惩罚异常值......有没有办法将r2_scorefromsklearn作为 LightGBM 的损失函数?

1个回答

R2 只是均方误差的重新缩放,LightGBM的默认损失函数所以像往常一样运行。(您可以使用另一个内置损失(MAE 或 Huber 损失?)来减少对异常值的惩罚。)