search

Ekeany / Boruta-Shap

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

[FEATURE] Add a transform method #81

Open AlvaroMatsuda opened 2 years ago

AlvaroMatsuda commented 2 years ago

Description

I was trying to chain BorutaShap inside a sklearn pipeline, as I can do with Borutapy, but it retuns this error:

TypeError: All intermediate steps should be transformers and implement fit and transform or be the string 'passthrough' '<BorutaShap.BorutaShap object at 0x000002038843DB20>' (type <class 'BorutaShap.BorutaShap'>) doesn't.

It would be nice if we could chain BorutaShap inside a sklearn pipeline as a feature selector.

Implementation

Maybe you could try to implement similar to the BorutaPy transform method?

ddofer commented 2 years ago

+1