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

show class counts when printing figs #114

Closed keyan3 closed 2 years ago

keyan3 commented 2 years ago

Adds a method print_tree that takes X and y, and shows the class proportions inside each node when printing FIGS.

Right now assumes classes are [0.0, 1.0] so this could be made more general.