stochtree0.3.0 package

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

createCppJsonFile

Create CppJson Object from File

createCppJsonString

Create CppJson Object from String

createCppRNG

Create CppRNG Object

createForest

Create Forest Object

createPreprocessorFromJsonString

Reload Covariate Preprocessor from JSON String

extract_parameter

Extract Parameter Samples Generic Function

Forest

Forest C++ Wrapper

ForestDataset

Forest Dataset C++ Wrapper

ForestModel

Forest Model C++ Wrapper

ForestModelConfig

Forest Model Configuration Object

ForestSamples

Forest Container C++ Wrapper

getRandomEffectSamples.bartmodel

Extract Random Effects Samples

getRandomEffectSamples.bcfmodel

Extract Random Effect Samples from BCF Model

getRandomEffectSamples

Extract Random Effect Samples Generic Function

loadVectorJson

Load Vector from JSON

Outcome

Outcome Data C++ Wrapper

plot.bartmodel

Plot BART Model Fit.

plot.bcfmodel

Plot BCF Model

saveBARTModelToJson

Convert BART Model to JSON

saveBARTModelToJsonFile

Save BART Model to JSON File

saveBARTModelToJsonString

Convert BART Model to JSON String

saveBCFModelToJson

Convert BCF Model to JSON

compute_contrast_bcf_model

Compute Contrast for BCF Model

computeForestLeafIndices

Query Forest Leaf Indices

createPreprocessorFromJson

Reload Covariate Preprocessor from JSON String

bart

Run BART for Supervised Learning

bcf

Run BCF for Causal Effect Estimation

calibrateInverseGammaErrorVariance

Calibrate Inverse Gamma Prior

compute_bart_posterior_interval

Compute BART Posterior Credible Intervals

compute_bcf_posterior_interval

Compute BCF Posterior Credible Intervals

compute_contrast_bart_model

Compute Contrast for BART Model

computeForestLeafVariances

Query Forest Leaf Scale Parameters

computeForestMaxLeafIndex

Query Forest Max Leaf Index

convertPreprocessorToJson

Convert Covariate Preprocessor to CppJson

CppJson

JSON C++ Object Wrapper

CppRNG

Random Number Generator C++ Wrapper

createBARTModelFromCombinedJson

Convert JSON List to Single BART Model

createBARTModelFromCombinedJsonString

Convert JSON String List to Single BART Model

createBARTModelFromJson

Convert JSON to BART Model

createBARTModelFromJsonFile

Convert JSON File to BART Model

createBARTModelFromJsonString

Convert JSON String to BART Model

createBCFModelFromCombinedJson

Convert JSON List to BCF Model

createBCFModelFromCombinedJsonString

Convert JSON String List to BCF Model

createBCFModelFromJson

Convert JSON to BCF Model

createBCFModelFromJsonFile

Convert JSON File to BCF Model

createBCFModelFromJsonString

Convert JSON String to BCF Model

createCppJson

Create CppJson Object

createForestDataset

Create ForestDataset Object

createForestModel

Create ForestModel Object

createForestModelConfig

Create ForestModelConfig Object

createForestSamples

Create ForestSamples Object

createGlobalModelConfig

Create GlobalModelConfig Object

createOutcome

Create Outcome Object

createRandomEffectSamples

Create RandomEffectSamples Object

createRandomEffectsDataset

Create RandomEffectsDataset Object

createRandomEffectsModel

Create RandomEffectsModel Object

createRandomEffectsTracker

Create RandomEffectsTracker Object

extract_parameter.bartmodel

Extract BART Parameter Samples.

extract_parameter.bcfmodel

Extract BCF Parameter Samples

GlobalModelConfig

Global Model Configuration Object

loadForestContainerCombinedJson

Combine JSON Model Objects into ForestSamples

loadForestContainerCombinedJsonString

Combine JSON Strings into ForestSamples

loadForestContainerJson

Load Forest Samples from JSON

loadRandomEffectSamplesCombinedJson

Combine JSON Model Objects into RandomEffectSamples

loadRandomEffectSamplesCombinedJsonString

Combine JSON Strings into RandomEffectSamples

loadRandomEffectSamplesJson

Load Random Effect Samples from JSON

loadScalarJson

Load Scalar from JSON

predict.bartmodel

Predict From a BART Model

predict.bcfmodel

Predict From BCF Model

preprocessPredictionData

Preprocess Covariates for Model Prediction

preprocessTrainData

Preprocess Covariates for Model Training

print.bartmodel

Summarize a BART Model

print.bcfmodel

Print Summary of BCF Model

RandomEffectSamples

Random Effect Container C++ Wrapper

RandomEffectsDataset

Random Effects Dataset C++ Wrapper

RandomEffectsModel

Random Effects Model C++ Wrapper

RandomEffectsTracker

Random Effects Tracker C++ Wrapper

resetActiveForest

Reset Active Forest

resetForestModel

Reset Forest Model

resetRandomEffectsModel

Reset RandomEffectsModel Object

resetRandomEffectsTracker

Reset RandomEffectsTracker Object

rootResetRandomEffectsModel

Reset RandomEffectsModel Object to Default State

rootResetRandomEffectsTracker

Reset RandomEffectsTracker Object to Default State

sample_bart_posterior_predictive

Sample BART Posterior Predictive

sample_bcf_posterior_predictive

Sample BCF Posterior Predictive

sample_without_replacement

Sample Without Replacement

sampleGlobalErrorVarianceOneIteration

Sample Global Error Variance

sampleLeafVarianceOneIteration

Sample Leaf Scale

saveBCFModelToJsonFile

Save BCF Model to JSON File

saveBCFModelToJsonString

Convert BCF Model to JSON String

savePreprocessorToJsonString

Convert Covariate Preprocessor to JSON String

stochtree-package

stochtree: Stochastic Tree Ensembles (XBART and BART) for Supervised L...

summary.bartmodel

Summarize the BART model fit and sampled terms.

summary.bcfmodel

Summarize BCF Model

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.

  • Maintainer: Drew Herren
  • License: MIT + file LICENSE
  • Last published: 2026-01-28