init

recommend-model-produce/src/main/java/com/tzld/piaoquan/recommend/model/produce/service/XGBoostService.java

@@ -90,6 +90,7 @@ public class XGBoostService {
 
             // 显示预测结果
             XGBoostClassificationModel model2 = XGBoostClassificationModel.load("file://" + destDir);
+            model2.setMissing(0.0f);
             Dataset<Row> predictions = model2.transform(assembledData);
             predictions.select("label", "prediction", "rawPrediction", "probability", "features").show(500);
 

recommend-model-produce/src/main/java/com/tzld/piaoquan/recommend/model/produce/xgboost/XGBoostTrainLocalTest.java

@@ -55,6 +55,7 @@ public class XGBoostTrainLocalTest {
 
             // 显示预测结果
             Dataset<Row> predictData = dataset("/Users/dingyunpeng/Desktop/part-00099.gz");
+            model.setMissing(0.0f);
             Dataset<Row> predictions = model.transform(predictData);
             predictions.select("label", "prediction", "rawPrediction", "probability", "features").show();