init

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

@@ -182,7 +182,7 @@ public class XGBoostService {
                 .setInputCols(features)
                 .setOutputCol("features");
 
-        Dataset<Row> assembledData = assembler.transform(dataset).select("features", "label");
+        Dataset<Row> assembledData = assembler.transform(dataset);
         assembledData.show();
         return assembledData;
     }