This directory contains replication code for
-
Athey and Wager (2019):
acic18 -
Athey, Tibshirani, and Wager (2019):
aos19 -
Cui, Kosorok, Sverdrup, Wager, and Zhu (2023):
csf -
Friedberg, Tibshirani, Athey, and Wager (2020):
local_linear_examples -
Mayer, Sverdrup, Gauss, Moyer, Wager, and Josse (2020): This is available at https://github.com/imkemayer/causal-inference-missing
-
Sverdrup, Petukhova, and Wager (2025):
ijmpr -
Wager and Athey (2018): This paper is not based on GRF, but on the deprecated
causalForest. For replication code see https://github.com/swager/causalForest -
Yadlowsky, Fleming, Shah, Brunskill, and Wager (2025): The method is available in the GRF function
rank_average_treatment_effect. For replication code see https://github.com/som-shahlab/RATE-experiments
Susan Athey and Stefan Wager. Estimating Treatment Effects with Causal Forests: An Application. Observational Studies, 5, 2019. [paper, arxiv]
Susan Athey, Julie Tibshirani and Stefan Wager. Generalized Random Forests. Annals of Statistics, 47(2), 2019. [paper, arxiv]
Yifan Cui, Michael R. Kosorok, Erik Sverdrup, Stefan Wager, and Ruoqing Zhu. Estimating Heterogeneous Treatment Effects with Right-Censored Data via Causal Survival Forests. Journal of the Royal Statistical Society: Series B, 85(2), 2023. [paper, arxiv]
Rina Friedberg, Julie Tibshirani, Susan Athey, and Stefan Wager. Local Linear Forests. Journal of Computational and Graphical Statistics, 30(2), 2020. [paper, arxiv]
Imke Mayer, Erik Sverdrup, Tobias Gauss, Jean-Denis Moyer, Stefan Wager and Julie Josse. Doubly Robust Treatment Effect Estimation with Missing Attributes. Annals of Applied Statistics, 14(3) 2020. [paper, arxiv]
Erik Sverdrup, Maria Petukhova, and Stefan Wager. Estimating Treatment Effect Heterogeneity in Psychiatry: A Review and Tutorial with Causal Forests. International Journal of Methods in Psychiatric Research, 34(2), 2025. [paper, arxiv]
Stefan Wager and Susan Athey. Estimation and Inference of Heterogeneous Treatment Effects using Random Forests. Journal of the American Statistical Association, 113(523), 2018. [paper, arxiv]
Steve Yadlowsky, Scott Fleming, Nigam Shah, Emma Brunskill, and Stefan Wager. Evaluating Treatment Prioritization Rules via Rank-Weighted Average Treatment Effects. Journal of the American Statistical Association, 120(549), 2025. [paper, arxiv]