imodels icon indicating copy to clipboard operation
imodels copied to clipboard

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

Open mattheweplandKH opened this issue 3 years ago • 1 comments

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.

mattheweplandKH avatar Aug 11 '22 04:08 mattheweplandKH

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

mepland avatar Dec 30 '22 21:12 mepland