Description
Stochastic Tree Ensembles (XBART and BART) for Supervised Learning and Causal Inference.
Description
Flexible stochastic tree ensemble software. Robust implementations of Bayesian Additive Regression Trees (BART) Chipman, George, McCulloch (2010) <doi:10.1214/09-AOAS285> for supervised learning and Bayesian Causal Forests (BCF) Hahn, Murray, Carvalho (2020) <doi:10.1214/19-BA1195> for causal inference. Enables model serialization and parallel sampling and provides a low-level interface for custom stochastic forest samplers.
README.md
stochtree R package
Software for building stochastic tree ensembles (i.e. BART, XBART) for supervised learning and causal inference.
Getting started
stochtree can be installed from CRAN via
install.package("stochtree")
The development version of stochtree can be installed from github via
remotes::install_github("StochasticTree/stochtree", ref="r-dev")