丁云鹏 8 tháng trước cách đây
mục cha
commit
cf1d25a998

+ 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);
 
 
             // 训练模型