search

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

plot function for figs #113

Closed OmerRonen closed 2 years ago

OmerRonen commented 2 years ago

@yanshuotan @csinva I added a plot function for figs (both the regular and ext classes). I also added the example usage in the main of the respective files. Under the hood, there's a simplified sklearn tree class that is extracted from the figs root node.

csinva commented 2 years ago

Looks good to me - can you add matplotlib as a dependency in the setup.py?