소스 검색

scala train

zhangbo 8 달 전
부모
커밋
1717f9926b
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

@@ -68,7 +68,7 @@ object train_01_xgb_ad_20240808{
       .setFeaturesCol("features")
       .setLabelCol("label")
       .setNthread(1)
-      .setNumWorkers(32)
+      .setNumWorkers(22)
     val model = xgbClassifier.fit(xgbInput)