timmens / causal-forest

Implements the Causal Forest algorithm formulated in Athey and Wager (2018).
MIT License
65 stars 12 forks source link

Implement (clever) parallelization. #5

Open timmens opened 4 years ago

timmens commented 4 years ago

Goal: Parallelize the causal_forest at the right level.

How to do it: No clear solution in mind. Maybe parallel processes for trees and multi threading for observation (when predicting).

Janos95 commented 4 years ago

I second this. This would greatly speed up my workflow! Keep up the good work!