csinva / imodels

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

FIGS Tweaks 3 #142

Closed mepland closed 1 year ago

mepland commented 1 year ago

Added code to create leaf predictions dict for plotting with dtreeviz, see here.

csinva commented 1 year ago

Looks good to me!