komiya-atsushi / xgboost-predictor-java

Pure Java implementation of XGBoost predictor for online prediction tasks.
Apache License 2.0
339 stars 108 forks source link

Handle ntree_limit higher than number of trees in GBTree and Dart model in graceful way #35

Closed jbojar closed 5 years ago

jbojar commented 6 years ago

If ntree_limit is set too high it uses maximum available trees in GBTree and Dart predictors.

It resolves issue #34.

jbojar commented 6 years ago

I had to bump jdk version to 8 because of strange issues with travis and unsupported jdk7.

kretes commented 6 years ago

@komiya-atsushi is there a chance to merge that and get an official version?

komiya-atsushi commented 5 years ago

Thanks!

komiya-atsushi commented 5 years ago

@kretes I have just released xgboost-predictor 0.3.1