-
-
Notifications
You must be signed in to change notification settings - Fork 25.8k
DOC Add link to Early Stopping example in Gradient Boosting #27025
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
DOC Add link to Early Stopping example in Gradient Boosting #27025
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
You can also add the link to the parameters in the API docs, i.e. the docstring of the HistGradientBoosting classes.
@adrinjalali, ready to review. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Now we need to improve the example on early stopping 😄
I'm also thinking something similar |
@greyisbetter Feel free to open PRs improving this or other examples. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I noticed that the example only uses GradientBoostingClassifier
, so this example can only be linked from GradientBoostingClassifier
and GradientBoostingRegressor
@adrinjalali, Could you please review the recent changes when you have a moment? |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Otherwise LGTM.
``n_iter_no_change`` consecutive epochs. The split is stratified, | ||
except in a multilabel setting. | ||
If early stopping is False, then the training stops when the training | ||
loss does not improve by more than tol for n_iter_no_change consecutive | ||
passes over the training set. | ||
Only effective when solver='sgd' or 'adam'. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
non-related change to the PR, please revert.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thank you for pointing that out. I've reverted the non-related changes as per your request. Your guidance is greatly appreciated
…earn#27025) Co-authored-by: Adrin Jalali <adrin.jalali@gmail.com>
Towards #26927
This adds a reference to example of choosing fewer
n_estimators
inGradientBoostingClassifier
model using Early Stopping technique.