Quellcode durchsuchen

Update train_01_xgb_ad_20250104: do not count to save time

StrayWarrior vor 3 Monaten
Ursprung
Commit
1fb9c04b78

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