丁云鹏 преди 8 месеца
родител
ревизия
cf1d25a998
променени са 1 файла, в които са добавени 2 реда и са изтрити 2 реда
  1. 2 2
      recommend-model-produce/src/main/java/com/tzld/piaoquan/recommend/model/produce/xgboost/XGBoostTrain.java

+ 2 - 2
recommend-model-produce/src/main/java/com/tzld/piaoquan/recommend/model/produce/xgboost/XGBoostTrain.java

@@ -114,8 +114,8 @@ public class XGBoostTrain {
                     .setMaxDepth(5)
                     .setObjective("binary:logistic")
                     .setNthread(1)
-                    .setNumRound(10)
-                    .setNumWorkers(2);
+                    .setNumRound(5)
+                    .setNumWorkers(1);
 
 
             // 训练模型