zhangbo преди 8 месеца
родител
ревизия
b17f3751e6
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      recommend-model-produce/src/main/scala/com/tzld/piaoquan/recommend/model/train_01_xgb_ad_20240808.scala

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

@@ -67,7 +67,7 @@ object train_01_xgb_ad_20240808{
       .setEvalMetric("auc")
       .setFeaturesCol("features")
       .setLabelCol("label")
-      .setNthread(8)
+      .setNthread(1)
       .setNumWorkers(1)
     val model = xgbClassifier.fit(xgbInput)