|
@@ -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);
|