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