init

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

@@ -38,7 +38,7 @@ public class XGBoostTrain {
             // 将 RDD[LabeledPoint] 转换为 JavaRDD<Row>
             JavaRDD<Row> rowRDD = rdd.map(s -> {
                 String[] line = StringUtils.split("\t");
-                String label = line[0];
+                int label = NumberUtils.toInt(line[0]);
                 double[] values = new double[line.length - 1];
                 for (int i = 1; i < line.length; i++) {
                     String[] fv = StringUtils.split(":");