search

csinva / imodels

Interpretable ML package 🔍 for concise, transparent, and accurate predictive modeling (sklearn-compatible).
https://csinva.io/imodels
MIT License
1.38k stars 122 forks source link

Create mean decrease in impurity (MDI) feature importances for tree-based models #127

Open mattheweplandKH opened 2 years ago

mattheweplandKH commented 2 years ago

While it is great that the models in imodels are readily interpretable, it would be nice to have a MDI feature importance, i.e. Gini importance, for tree-based models like FIGS to compare to other tree-based models like sklearn's RandomForestClassifier and GradientBoostingClassifier.

mepland commented 1 year ago

Added for FIGS in https://github.com/csinva/imodels/pull/154