-
Notifications
You must be signed in to change notification settings - Fork 40
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
ShapRFECV extend support for regression and multiclass #169
Comments
I've implemented this for a project I'm working on. It works for LightGBM multi-class classification and regression. I haven't tested it on other models but I don't expect there would be any issues. I will make a pull request if you want to review it? |
That would be great! One request, if you make unit tests for lightgbm please add the following decorator to make sure that these tests are not executed on mac OS, since we currently have a problem with it.
|
#169 support added for multi-class classification. Next to that unit test added, a bug fixed and spurious code in tests removed. --------- Co-authored-by: Luke Rothwell <[email protected]> Co-authored-by: Reinier Koops <[email protected]>
fixed in #248 |
…, removed unused code and updated notebooks to work (again). (#248) This PR depends on the PR to be accepted: #242 ______ The PR fixes the following: - Removed unused code and so partially fixed: #245 - Updates all notebooks so they are all working: #246 - Add explicit support for regressors and multiclass: : #241 & : #169 - Updated the yaml files for github actions. Now we have a weekly cronjob (instead of daily) to test the notebooks. - Enable tests which were previously failing or only enabled in certain circumstances (much has changed the last 3 years with regards of adoption of the Mac ARM architecture support) - Removed most (if not all except one on purpose) of the copyright notice in code. --------- Signed-off-by: dependabot[bot] <[email protected]> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Problem Description
Currently ShapRFECV supports only binary classification. It most probably works fine for other types of models.
Desired Outcome
The text was updated successfully, but these errors were encountered: