Browse Source

cold start

丁云鹏 8 months ago
parent
commit
04424e8bfd

+ 1 - 1
recommend-model-produce/src/main/java/com/tzld/piaoquan/recommend/model/produce/xgboost/XGBoostTrain.java

@@ -39,7 +39,7 @@ public class XGBoostTrain {
                 }
                 return RowFactory.create(label, values);
             });
-
+            log.info("rowRDD count {}", rowRDD.count());
             // 将 JavaRDD<Row> 转换为 Dataset<Row>
             Dataset<Row> dataset = spark.createDataFrame(rowRDD, Row.class);