gpt4 book ai didi

python - 为回归重现 LightGBM 自定义损失函数

转载 作者:行者123 更新时间:2023-12-03 19:17:03 31 4
gpt4 key购买 nike

我想重现 LightGBM 的自定义损失函数。这是我尝试过的:

lgb.train(params=params, train_set=dtrain, num_boost_round=num_round, fobj=default_mse_obj)

default_mse_obj 被定义为:
residual = y_true - y_pred.get_label()
grad = -2.0*residual
hess = 2.0+(residual*0)
return grad, hess

但是,与定义的自定义损失函数相比,默认“回归”目标的 eval 指标是不同的。我想知道,LightGBM 用于“回归”目标的默认函数是什么?

最佳答案

如您所见 here ,这是回归任务的默认损失函数

def default_mse_obj(y_pred, dtrain):

y_true = dtrain.get_label()

grad = (y_pred - y_true)
hess = np.ones(len(grad))

return grad, hess

关于python - 为回归重现 LightGBM 自定义损失函数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/60290304/

31 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com