Bladeren bron

feat:修改批量更新创意属性接口

zhaohaipeng 8 maanden geleden
bovenliggende
commit
853d3b18b8
1 gewijzigde bestanden met toevoegingen van 6 en 2 verwijderingen
  1. 6 2
      src/main/scala/com/aliyun/odps/spark/ad/xgboost/v20240808/XGBoostTrain.scala

+ 6 - 2
src/main/scala/com/aliyun/odps/spark/ad/xgboost/v20240808/XGBoostTrain.scala

@@ -76,14 +76,18 @@ object XGBoostTrain {
 
       // 创建 XGBoostClassifier 对象
       val xgbClassifier = new XGBoostClassifier()
-        .setEta(0.1f)
+        .setEta(0.01f)
+        .setSubsample(0.8)
+        .setColsampleBytree(0.8)
+        .setScalePosWeight(1)
+        .setSeed(2024)
         .setMissing(0.0f)
         .setFeaturesCol("features")
         .setLabelCol("label")
         .setMaxDepth(5)
         .setObjective("binary:logistic")
         .setNthread(1)
-        .setNumRound(5)
+        .setNumRound(100)
         .setNumWorkers(1)
 
       // 训练模型