boruta_py icon indicating copy to clipboard operation
boruta_py copied to clipboard

select features & xgboost with Shap

Open gdupret opened this issue 3 years ago • 0 comments

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 avatar May 26 '21 00:05 gdupret