stochtree: Stochastic Tree Ensembles (XBART and BART) for Supervised Learning and Causal Inference
Andrew Herren, Richard Hahn, Jared S. Murray, Carlos Carvalho, Jingyu He · 2025
Flexible stochastic tree ensemble software. Robust implementations of Bayesian Additive Regression Trees (BART) Chipman, George, McCulloch (2010) for supervised learning and Bayesian Causal Forests (BCF) Hahn, Murray, Carvalho (2020) for causal inference. Enables model serialization and parallel sampling and provides a low-level interface for custom stochastic forest samplers.