[MRG+1] add groups support to RFECV#9656
Merged
TomDLT merged 9 commits intoscikit-learn:masterfrom Nov 17, 2017
Merged
Conversation
Member
|
Needs a test. |
jnothman
reviewed
Sep 1, 2017
| X = iris.data | ||
| y = (iris.target > 0).astype(int) | ||
|
|
||
| # test basic cv |
Member
There was a problem hiding this comment.
Why is this needed? Isn't it tested elsewhere?
Contributor
Author
Member
|
otherwise LGTM |
jnothman
reviewed
Sep 2, 2017
| assert_array_almost_equal(preds_sparse, preds) | ||
|
|
||
|
|
||
| def test_rfe_cv(): |
Member
There was a problem hiding this comment.
Silly me, this shouldn't be in test_cross_validation. It should be in feature_selection/tests/test_rfe.py
glemaitre
reviewed
Sep 5, 2017
| cv=GroupKFold(n_splits=2) | ||
| ) | ||
| est_groups.fit(X, y, groups=groups) | ||
| assert(est_groups.n_features_ > 0) |
Member
There was a problem hiding this comment.
You don't need the paranthesis
|
|
||
|
|
||
| def test_rfe_cv_groups(): | ||
| # test cv with groups |
Member
There was a problem hiding this comment.
you can remove the comment since this is already implied by the name of the function
| y = (iris.target > 0).astype(int) | ||
|
|
||
| est_groups = RFECV( | ||
| estimator=RandomForestClassifier(), |
Member
There was a problem hiding this comment.
could you set the random_state of the estimator that take one.
Member
|
Good job @adamgreenhall, thanks ! |
jwjohnson314
pushed a commit
to jwjohnson314/scikit-learn
that referenced
this pull request
Dec 18, 2017
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
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
4 participants
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 Issue
related to #8127
What does this implement/fix? Explain your changes.
Adding support for cv groups (eg
LeaveOneGroupOut) toRFECV.