stochtree: Stochastic Tree Ensembles (XBART and BART) for Supervised Learning and Causal Inference

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.

Version: 0.1.1
Imports: R6, stats
LinkingTo: cpp11, BH
Suggests: testthat (≥ 3.0.0)
Published: 2025-02-08
DOI: 10.32614/CRAN.package.stochtree
Author: Drew Herren ORCID iD [aut, cre], Richard Hahn [aut], Jared Murray [aut], Carlos Carvalho [aut], Jingyu He [aut], Pedro Lima [ctb], stochtree contributors [cph], Eigen contributors [cph] (C++ source uses the Eigen library for matrix operations, see inst/COPYRIGHTS), xgboost contributors [cph] (C++ tree code and related operations include or are inspired by code from the xgboost library, see inst/COPYRIGHTS), treelite contributors [cph] (C++ tree code and related operations include or are inspired by code from the treelite library, see inst/COPYRIGHTS), Microsoft Corporation [cph] (C++ I/O and various project structure code include or are inspired by code from the LightGBM library, which is a copyright of Microsoft, see inst/COPYRIGHTS), Niels Lohmann [cph] (C++ source uses the JSON for Modern C++ library for JSON operations, see inst/COPYRIGHTS), Daniel Lemire [cph] (C++ source uses the fast_double_parser library internally, see inst/COPYRIGHTS), Victor Zverovich [cph] (C++ source uses the fmt library internally, see inst/COPYRIGHTS)
Maintainer: Drew Herren <drewherrenopensource at gmail.com>
BugReports: https://github.com/StochasticTree/stochtree/issues
License: MIT + file LICENSE
Copyright: Copyright details for stochtree's C++ dependencies, which are vendored along with the core stochtree source code, are detailed in inst/COPYRIGHTS
stochtree copyright details
URL: https://stochtree.ai/, https://github.com/StochasticTree/stochtree
NeedsCompilation: yes
SystemRequirements: C++17
Materials: README NEWS
CRAN checks: stochtree results

Documentation:

Reference manual: stochtree.pdf

Downloads:

Package source: stochtree_0.1.1.tar.gz
Windows binaries: r-devel: stochtree_0.1.0.zip, r-release: stochtree_0.1.1.zip, r-oldrel: stochtree_0.1.0.zip
macOS binaries: r-release (arm64): stochtree_0.1.1.tgz, r-oldrel (arm64): not available, r-release (x86_64): stochtree_0.1.1.tgz, r-oldrel (x86_64): not available
Old sources: stochtree archive

Linking:

Please use the canonical form https://CRAN.R-project.org/package=stochtree to link to this page.