spark xgboost & lightgbm 参数解释

一、spark xgboost 模型

1 xgboost 的默认参数:

xgb 参数参考链接 https://blog.csdn.net/yyy430/article/details/85179638 这个链接整理的比较全,但是这个参数是关于python版本的xgb,spark版本的xgboost默认参数和这个有出入

1.1 默认参数如下:

  /*
默认参数
	eta -> 0.3
   , gamma -> 0
   , maxDepth -> 6,
    minChildWeight -> 1
    ,maxDeltaStep -> 0,
    growPolicy -> "depthwise"
    ,maxBins -> 16,// python默认是256
    subsample -> 1
    ,colsampleBytree -> 1
    ,colsampleBylevel -> 1
    ,lambda -> 1
    ,alpha -> 0
    ,treeMethod -> "auto"
    ,sketchEps -> 0.03,
   scalePosWeight -> 1.0
   ,sampleType -> "uniform"
   ,normalizeType -> "tree"
    ,rateDrop -> 0.0
    ,skipDrop -> 0.0
    ,lambdaBias -> 0
    ,treeLimit -> 0
             */

2 、spark xgb 模型的参数


            
            val paraMap = List(
                  //参数解释 https://blog.csdn.net/Leo_Sheng/article/details/80852328
                  "eta" -> 0.3f // learning rate
                  ,"gamma" -> 0.1 //用于控制是否后剪枝的参数,越大越保守,一般0.1、0.2这样子。
                  ,"max_depth" -> max_depth
                  ,"num_round"->max_iter
                  ,"objective" -> "binary:logistic"
                  
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值