Ekeany / Boruta-Shap

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

[ENH] Improved param search compatibility #121

Open matheusvasconcellosb opened 1 year ago

matheusvasconcellosb commented 1 year ago

Current Situation

The class works well if used in isolation but not that much inside a pipeline.

Enhancement

We could clean up fit method and move params to class definition to be easier to perform parameter tunning using scikit-learn compatible stuff..

Reasoning

Why should this enhancement be added to the repository? --> scikit-learn better compatibility

Implementation

move paramters to class definition

Tasks

Concrete tasks to be completed, in the order they need to be done in. Include links to specific lines of code where the task should happen at.