search

csinva / imodels

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

rulefit cv optionality #73

Closed keyan3 closed 2 years ago

keyan3 commented 2 years ago

Rulefit uses cross-validation on the training set to pick the best alpha that satisfies max_rules — added a cv arg to make it possible to disable this to speed things up.

Also search over 100 alphas for both regression and classification (previously was 100 for regression but 10 for classification)