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