Ekeany / Boruta-Shap

A Tree based feature selection tool which combines both the Boruta feature selection algorithm with shapley values.
MIT License
559 stars 86 forks source link

MIssing import for inspect #120

Open matheusvasconcellosb opened 9 months ago

matheusvasconcellosb commented 9 months ago

https://github.com/Ekeany/Boruta-Shap/blob/cddb6da69c59b58b9e4d74681202fb0d08ec54f7/src/BorutaShap.py#L78C29-L78C29

Hey, it is missing an import here for get_params to work. Also if you subclass BaseEstimator/TransformerMixin you can get rid of get_params/set_params etc..