csinva / imodels

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

add rf n_estimators param #47

Closed keyan3 closed 3 years ago

keyan3 commented 3 years ago

RuleFit uses max_rules during extraction to determine how many trees to fit, then uses it again during scoring to limit the max number of final rules. It probably makes more sense to accept a dedicated n_estimators parameter to accomplish the former