|
@@ -124,7 +124,7 @@ public class XGBoostTrain {
|
|
|
|
|
|
// 显示预测结果
|
|
// 显示预测结果
|
|
Dataset<Row> predictions = model.transform(testData);
|
|
Dataset<Row> predictions = model.transform(testData);
|
|
- predictions.select("label", "prediction").show(1000);
|
|
|
|
|
|
+ predictions.select("label", "prediction").show(10000);
|
|
|
|
|
|
|
|
|
|
} catch (Throwable e) {
|
|
} catch (Throwable e) {
|