@@ -50,7 +50,7 @@ public class XGBRegressionModel extends Model {
values[i] = v;
}
DMatrix dm = new DMatrix(values, 1, features.length, 0.0f);
- float[][] result = model._booster().predict(dm, false, 100);
+ float[][] result = model._booster().predict(dm, false, 0);
return result[0][0];
} catch (Exception e) {
return 0f;
@@ -52,7 +52,7 @@ public class XGBoostModel extends Model {