瀏覽代碼

scala train

zhangbo 8 月之前
父節點
當前提交
b17f3751e6

+ 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)