-
Notifications
You must be signed in to change notification settings - Fork 39
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
Multi class shap rfecv #174
Conversation
multi-class classifiers and regressors are now compatible, so we remove references to binary classification
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 working on this, and extending the functionality for other users.
I left some comments to implement and questions, but overall looks good.
Also please have a look what has changed in, just to make sure your PR is aligned #175 |
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.
Looks good, thanks for the work and sorry for the delay.
If you fix the conflicts and make the test sucessfully run we can directly merge ;) Keep in mind that there are 3 new functions added for EarlyStoppingRFECV that pass early stopping parameters differently based on model type: LGBM/XGB/CatBoost
Hi @Matgrb, can we merge this? |
Hi @lproth , sorry for the delayed response and thank you for your contribution, could you update this branch to be compatible with the latest changes on the main? |
This can only be merged when some unit tests are added. |
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.
Update has been approved in previous commits and now just updates to be compatible with main
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.
Fixed some bugs & added a unit test. removed spurious code.
Re #169 I have added support for multi-class classification and ensured that regression will still work. I haven't written any unit tests because I'm not familiar with how pytest works, but all that should be needed is to re-run the existing tests with a multi-class target instead of binary.