TeamHG-Memex / eli5

A library for debugging/inspecting machine learning classifiers and explaining their predictions
http://eli5.readthedocs.io
MIT License
2.74k stars 332 forks source link

pass groups of features eli5.sklearn.permutation_importance #341

Open omarcr opened 4 years ago

omarcr commented 4 years ago

HI, I would like to pass groups of features to Eli to permute such groups in order to prevent to permute single features (highly computational expensive). Is there a way to do this?

For example if I have: features= [1, 2, 3, 4, 5, 6]

and I want to pass the following group to Eli features_group = [A, A, A, B, B, B]

in order to permute groups not features itself to calculate permutation_importance, is this possible?

Thanks