DOC Add warm start section for tree ensembles #29001
Merged
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
adrinjalali
reviewed
May 13, 2024
doc/modules/ensemble.rst
Outdated
| >>> X, y = make_classification(n_samples=100, random_state=1) | ||
| >>> clf = RandomForestClassifier(n_estimators=10) | ||
| >>> clf = clf.fit(X, y) # fit with 10 trees | ||
| >>> _ = clf.set_params(n_estimators=20, warm_start=True) # set warm_start and increase num of estimators |
There was a problem hiding this comment.
Line too long, might be a good idea to move the comment to the line above?
Also, do you think it makes sense to show something about the estimator before and after the second fit? To show how it changed?
There was a problem hiding this comment.
Good idea. I just printed len(clf.estimators_).
I guess ideally we would show that the first 10 estimators are the same but I couldn't think of a good way to do that.
adrinjalali
approved these changes
May 14, 2024
There was a problem hiding this comment.
We could export the trees with graphviz and show, but this is already a good improvement. Further improvements are very much welcome. Thanks @lucyleeow
jeremiedbb
pushed a commit
to jeremiedbb/scikit-learn
that referenced
this pull request
May 20, 2024
jeremiedbb
pushed a commit
that referenced
this pull request
May 21, 2024
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Reference Issues/PRs
closes #22041
What does this implement/fix? Explain your changes.
warm_start#24579), which is confusing. I thought about making a single section and talk about both ensemble and gradient boosted trees but thought it didn't fit withensemble.rstpage as its currently divided into sections for each estimator type.warm_starttogether withrandom_stateis poorly documented #22041). I thought about adding this section in the docstrings (withwarm_startparam or at the end) of the 6 estimators but thought this was better as we can give an example as well. Happy to change though.Any other comments?