丁云鹏 8 months ago
parent
commit
c964f53d1a

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

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