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

Parallelize HSTree application to fitted forest models #125

Open Innixma opened 2 years ago

Innixma commented 2 years ago

I looked at the inner code of HSTree and for forest models it is simply for looping through the trees and applying the reg_param update. This works, but will only use 1 CPU core. Sklearn when training these forest models uses multithreading/processing to speed things up considerably. Have the authors considered adding parallel support to HSTree for forest models?