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