Skip to content
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

select features & xgboost with Shap #97

Draft
wants to merge 3 commits into
base: master
Choose a base branch
from

Conversation

gdupret
Copy link

@gdupret gdupret commented May 26, 2021

Use Shap to evaluate the feature importance in XGBoost models

Shap is not available from the sklearn API to XGBoost so I added a wrapper around XGBoost to include it

Select the features to be tested

Sometimes some features are known to be important, so there is no benefit in testing them. Removing these features allows to save computing time and memory.

Miscellaneous

rewrite np.int to int, np.float to float and np.bool to bool to avoid deprecation warnings

@gdupret gdupret marked this pull request as draft May 27, 2021 00:15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant