丁云鹏 il y a 8 mois
Parent
commit
123a71f890

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

@@ -78,8 +78,8 @@ public class XGBoostTrain {
                     map.put(fv[0], NumberUtils.toDouble(fv[1], 0.0));
                     map.put(fv[0], NumberUtils.toDouble(fv[1], 0.0));
                 }
                 }
 
 
-                int[] indices = new int[features.length];
-                double[] values = new double[features.length];
+                int[] indices = new int[features.size()];
+                double[] values = new double[features.size()];
                 for (int i = 0; i < features.size(); i++) {
                 for (int i = 0; i < features.size(); i++) {
                     indices[i] = i;
                     indices[i] = i;
                     values[i] = map.getOrDefault(features.get(i), 0.0);
                     values[i] = map.getOrDefault(features.get(i), 0.0);