122 hstree multiclass classification support

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

Closed csinva closed 2 years ago

yanshuotan commented 2 years ago

Multi-class classification functionality added. Other small changes:

Fixed a bug where intermediate node values for classification were inaccurate
Changed the denominator in the regularization factor from n_node_samples to weighted_n_node_samples