瀏覽代碼

feat:修改XGB训练参数

ZhaoHP 4 月之前
父節點
當前提交
bb9b1054a8

+ 2 - 1
recommend-model-produce/src/main/scala/com/tzld/piaoquan/recommend/model/train_01_xgb_ad_20240808.scala

@@ -37,6 +37,7 @@ object train_01_xgb_ad_20240808{
     val func_object = param.getOrElse("func_object", "binary:logistic")
     val func_metric = param.getOrElse("func_metric", "auc")
     val repartition = param.getOrElse("repartition", "20").toInt
+    val subsample = param.getOrElse("subsample", "0.95").toDouble
 
     val loader = getClass.getClassLoader
     val resourceUrl = loader.getResource(featureFile)
@@ -84,7 +85,7 @@ object train_01_xgb_ad_20240808{
       .setMissing(0.0f)
       .setMaxDepth(max_depth)
       .setNumRound(num_round)
-      .setSubsample(0.8)
+      .setSubsample(subsample)
       .setColsampleBytree(0.8)
       .setScalePosWeight(1)
       .setObjective(func_object)