csinva / imodels

Interpretable ML package 🔍 for concise, transparent, and accurate predictive modeling (sklearn-compatible).

https://csinva.io/imodels

MIT License

1.38k stars 122 forks source link

Closed mcschmitz closed 1 year ago

mcschmitz commented 1 year ago

Changes

Update the fit_node_recursive method of the GreedyRuleListClassifier to propagate the y_left value as initial probabilty to the next lower level when fitting the model
Apply the flipped condition if a rule has to be flipped during fitting.
Remove unused imports from the touched files
Add tests for the fixes
Notes

This PR fixes #167

csinva commented 1 year ago

Looks great!