【干货】 xgboost如何自定义eval_metric ( feval ) ?

,#【干货】 xgboost如何自定义eval_metric ( feval ) ?

转载请注明原文地址:http://blog.csdn.net/weixin_38100489/article/details/78714251

问题来源:xgboost的eval_metric里如果没有你要用的,比如kaggle里有用到的rmsle就没有就需要自己定义。

http://xgboost.readthedocs.io/en/latest/parameter.html
官网截图

上代码:(把eval_metric变成rmsle的例子)

dtrain=xgb.DMatrix(train_x,train_y)
dtest=xgb.DMatrix(test_x,test_y_real)
watchlist=[(dtrain,'train'),(dtest,'test')]

#############
def evalerror(preds, dtrain):		# written by myself
    labels = dtrain.get_label()
    # return a pair metric_name, result
    # since preds are margin(before logistic transformation, cutoff at 0)
    return 'error', math.sqrt(mean_squared_log_error(preds,labels))
#############

params = {
            'objective': 'reg:gamma',
            'eta': 0.1,
            'seed': 0,
            #'eval_metric': 'rmse',        # if it is useless, delete it
            'missing': -999,
            'silent' : 1,
            'gamma' : 1,
            'subsample' : 0.5,
            'alpha' : 1,
            'max_depth':10,
            'min_child_weight':1
            }
num_rounds=500
clf=xgb.train(params,dtrain,num_rounds,watchlist, feval=evalerror)

这样就可以啦,关键就是‘feval=’,在xgb.train内的实参是叫feval,不是叫eval_metric !!!
(上面代码参数我都可以改花了)

xgb.train(params, dtrain, num_boost_round, evals, obj, feval, maximize, early_stopping_rounds, evals_result, verbose_eval, learning_rates, xgb_model, callbacks)

中的第六个。

思路来源是https://www.zhihu.com/question/39496618 上的大神 lau phunter 的回答,https://github.com/dmlc/xgboost/blob/master/demo/guide-python/custom_objective.py
———————————————————————————————————————————————————————
~一把辛酸泪,可费劲了~
我的知乎:https://www.zhihu.com/people/xu-jian-zhi/posts

  • 8
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值