TeamHG-Memex / eli5

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

explain_weights should support RFE/RFECV and other metaestimators #356

Open hermidalc opened 4 years ago

hermidalc commented 4 years ago

RFE/RFECV are not only feature selectors (SelectorMixin) but also classifiers/regressors (MetaEstimatorMixin), though ELI5 explainweights doesn't support them as classifiers/regressors. The final fit of an RFE/RFECV object is a fitted estimator with either `rfe.estimator.coeforrfe.estimator.featureimportances` and in sklearn you do not usually follow up RFE/RFECV with another classifier or regressor. Would be great if ELI5 could support metaestimators in explain_weights.