StochTree R API Reference#
Overview of the stochtree
R library's key classes and functions, built as a self-contained doc site using the pkgdown format. The stochtree
interface is divided into two "levels":
- "High level": end-to-end implementations of stochastic tree ensembles for supervised learning (BART / XBART) and causal inference (BCF / XBCF).
- "Low level": we provide access to most of the C++ sampling objects and functionality via R, which allow for custom sampling algorithms and integration of other model terms. This interface consists broadly of the following components:
- Data API: loading and storing in-memory data needed to train
stochtree
models. - Forest API: creating, storing, modifying, and sampling ensembles of decision trees that underlie all
stochtree
models. - Serialization API: serializing models to JSON (files or in-memory strings).
- Random Effects API: sampling from additive random effects models.
- Data API: loading and storing in-memory data needed to train