소스 검색

Update train_01_xgb_ad_20250104: do not count to save time

StrayWarrior 3 달 전
부모
커밋
064b612792
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      recommend-model-produce/src/main/scala/com/tzld/piaoquan/recommend/model/train_01_xgb_ad_20250104.scala

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

@@ -68,7 +68,7 @@ object train_01_xgb_ad_20250104 {
       features,
       negSampleRate
     )
-    println("path %s, train data size:%d".format(trainPath, trainData.count()))
+    // println("path %s, train data size:%d".format(trainPath, trainData.count()))
     val trainDataSet: Dataset[Row] = spark.createDataFrame(trainData, schema)
     val vectorAssembler = new VectorAssembler().setInputCols(features).setOutputCol("features")
     val xgbInput = vectorAssembler.transform(trainDataSet).select("features", "label")