소스 검색

cold start

丁云鹏 10 달 전
부모
커밋
04424e8bfd
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      recommend-model-produce/src/main/java/com/tzld/piaoquan/recommend/model/produce/xgboost/XGBoostTrain.java

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