-
-
Notifications
You must be signed in to change notification settings - Fork 8.7k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[feature ] return the best model after early stop #10816
Comments
I guess when predicting, you can specify how many trees to be used. But currently, xgboost4j has not supported specifying the tree limit. I will make a PR for it. |
Jvm doesn't support model slicing yet. @wbo4958 this might help
|
Cool, looks like it's doable. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Does xgboost4j-spark support to get the best model after early stop?
It seems it will get the model at that iteration which is best iteration + num_early_stopping_rounds, am i wrong?
how could i get the best model?
The text was updated successfully, but these errors were encountered: