Package stochtree updated to version 0.4.0 with previous version 0.3.1 dated 2026-02-12

Fri, 06 Mar 2026 03:02:00 CDT

Title: 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. Includes the grow-from-root algorithm for accelerated forest sampling (He and Hahn (2021) <doi:10.1080/01621459.2021.1942012>), a log-linear leaf model for forest-based heteroskedasticity (Murray (2020) <doi:10.1080/01621459.2020.1813587>), and the cloglog BART model of Alam and Linero (2025) <doi:10.48550/arXiv.2502.00606> for ordinal outcomes.
Author: Drew Herren [aut, cre] , Richard Hahn [aut], Jared Murray [aut], Carlos Carvalho [aut], Jingyu He [aut], Pedro Lima [ctb], Entejar Alam [ctb], stochtree contributors [cph], Eigen contributors [cph] , xgboost contributors [cph] , treelite contributors [...truncated...]
Maintainer: Drew Herren <drewherrenopensource@gmail.com>

Diff between stochtree versions 0.3.1 dated 2026-02-12 and 0.4.0 dated 2026-03-06

 stochtree-0.3.1/stochtree/man/computeForestLeafIndices.Rd                  |only
 stochtree-0.3.1/stochtree/man/computeForestLeafVariances.Rd                |only
 stochtree-0.3.1/stochtree/man/computeForestMaxLeafIndex.Rd                 |only
 stochtree-0.3.1/stochtree/man/compute_bart_posterior_interval.Rd           |only
 stochtree-0.3.1/stochtree/man/compute_bcf_posterior_interval.Rd            |only
 stochtree-0.3.1/stochtree/man/compute_contrast_bart_model.Rd               |only
 stochtree-0.3.1/stochtree/man/compute_contrast_bcf_model.Rd                |only
 stochtree-0.3.1/stochtree/man/convertPreprocessorToJson.Rd                 |only
 stochtree-0.3.1/stochtree/man/createBARTModelFromCombinedJson.Rd           |only
 stochtree-0.3.1/stochtree/man/createBARTModelFromCombinedJsonString.Rd     |only
 stochtree-0.3.1/stochtree/man/createBARTModelFromJson.Rd                   |only
 stochtree-0.3.1/stochtree/man/createBARTModelFromJsonFile.Rd               |only
 stochtree-0.3.1/stochtree/man/createBARTModelFromJsonString.Rd             |only
 stochtree-0.3.1/stochtree/man/createBCFModelFromCombinedJson.Rd            |only
 stochtree-0.3.1/stochtree/man/createBCFModelFromCombinedJsonString.Rd      |only
 stochtree-0.3.1/stochtree/man/createBCFModelFromJson.Rd                    |only
 stochtree-0.3.1/stochtree/man/createBCFModelFromJsonFile.Rd                |only
 stochtree-0.3.1/stochtree/man/createBCFModelFromJsonString.Rd              |only
 stochtree-0.3.1/stochtree/man/createCppJson.Rd                             |only
 stochtree-0.3.1/stochtree/man/createCppJsonFile.Rd                         |only
 stochtree-0.3.1/stochtree/man/createCppJsonString.Rd                       |only
 stochtree-0.3.1/stochtree/man/createPreprocessorFromJson.Rd                |only
 stochtree-0.3.1/stochtree/man/createPreprocessorFromJsonString.Rd          |only
 stochtree-0.3.1/stochtree/man/extract_parameter.Rd                         |only
 stochtree-0.3.1/stochtree/man/extract_parameter.bartmodel.Rd               |only
 stochtree-0.3.1/stochtree/man/extract_parameter.bcfmodel.Rd                |only
 stochtree-0.3.1/stochtree/man/loadForestContainerCombinedJson.Rd           |only
 stochtree-0.3.1/stochtree/man/loadForestContainerCombinedJsonString.Rd     |only
 stochtree-0.3.1/stochtree/man/loadForestContainerJson.Rd                   |only
 stochtree-0.3.1/stochtree/man/loadRandomEffectSamplesCombinedJson.Rd       |only
 stochtree-0.3.1/stochtree/man/loadRandomEffectSamplesCombinedJsonString.Rd |only
 stochtree-0.3.1/stochtree/man/loadRandomEffectSamplesJson.Rd               |only
 stochtree-0.3.1/stochtree/man/preprocessPredictionData.Rd                  |only
 stochtree-0.3.1/stochtree/man/preprocessTrainData.Rd                       |only
 stochtree-0.3.1/stochtree/man/resetActiveForest.Rd                         |only
 stochtree-0.3.1/stochtree/man/resetForestModel.Rd                          |only
 stochtree-0.3.1/stochtree/man/resetRandomEffectsModel.Rd                   |only
 stochtree-0.3.1/stochtree/man/resetRandomEffectsTracker.Rd                 |only
 stochtree-0.3.1/stochtree/man/rootResetRandomEffectsModel.Rd               |only
 stochtree-0.3.1/stochtree/man/rootResetRandomEffectsTracker.Rd             |only
 stochtree-0.3.1/stochtree/man/sample_bart_posterior_predictive.Rd          |only
 stochtree-0.3.1/stochtree/man/sample_bcf_posterior_predictive.Rd           |only
 stochtree-0.3.1/stochtree/man/saveBARTModelToJson.Rd                       |only
 stochtree-0.3.1/stochtree/man/saveBARTModelToJsonFile.Rd                   |only
 stochtree-0.3.1/stochtree/man/saveBARTModelToJsonString.Rd                 |only
 stochtree-0.3.1/stochtree/man/saveBCFModelToJson.Rd                        |only
 stochtree-0.3.1/stochtree/man/saveBCFModelToJsonFile.Rd                    |only
 stochtree-0.3.1/stochtree/man/saveBCFModelToJsonString.Rd                  |only
 stochtree-0.3.1/stochtree/man/savePreprocessorToJsonString.Rd              |only
 stochtree-0.4.0/stochtree/DESCRIPTION                                      |   16
 stochtree-0.4.0/stochtree/MD5                                              |  244 -
 stochtree-0.4.0/stochtree/NAMESPACE                                        |   24
 stochtree-0.4.0/stochtree/NEWS.md                                          |   34
 stochtree-0.4.0/stochtree/R/bart.R                                         | 1237 +++++++---
 stochtree-0.4.0/stochtree/R/bcf.R                                          |  888 ++-----
 stochtree-0.4.0/stochtree/R/calibration.R                                  |    6
 stochtree-0.4.0/stochtree/R/config.R                                       |   56
 stochtree-0.4.0/stochtree/R/cpp11.R                                        |   36
 stochtree-0.4.0/stochtree/R/data.R                                         |   80
 stochtree-0.4.0/stochtree/R/forest.R                                       |  215 +
 stochtree-0.4.0/stochtree/R/generics.R                                     |   10
 stochtree-0.4.0/stochtree/R/kernel.R                                       |   96
 stochtree-0.4.0/stochtree/R/model.R                                        |   21
 stochtree-0.4.0/stochtree/R/posterior_transformation.R                     |  376 ++-
 stochtree-0.4.0/stochtree/R/random_effects.R                               |  320 +-
 stochtree-0.4.0/stochtree/R/serialization.R                                |  297 +-
 stochtree-0.4.0/stochtree/R/stochtree-package.R                            |    1
 stochtree-0.4.0/stochtree/R/utils.R                                        |  210 +
 stochtree-0.4.0/stochtree/R/variance.R                                     |    8
 stochtree-0.4.0/stochtree/configure                                        |   18
 stochtree-0.4.0/stochtree/configure.ac                                     |    2
 stochtree-0.4.0/stochtree/man/BARTSerialization.Rd                         |only
 stochtree-0.4.0/stochtree/man/BCFSerialization.Rd                          |only
 stochtree-0.4.0/stochtree/man/CppJson.Rd                                   |    2
 stochtree-0.4.0/stochtree/man/DataPreprocessing.Rd                         |only
 stochtree-0.4.0/stochtree/man/Forest.Rd                                    |   12
 stochtree-0.4.0/stochtree/man/ForestDataset.Rd                             |  113
 stochtree-0.4.0/stochtree/man/ForestKernelComputation.Rd                   |only
 stochtree-0.4.0/stochtree/man/ForestModel.Rd                               |    2
 stochtree-0.4.0/stochtree/man/ForestModelConfig.Rd                         |   74
 stochtree-0.4.0/stochtree/man/ForestSamples.Rd                             |   12
 stochtree-0.4.0/stochtree/man/ForestSamplesSerialization.Rd                |only
 stochtree-0.4.0/stochtree/man/ForestStateManagement.Rd                     |only
 stochtree-0.4.0/stochtree/man/GlobalModelConfig.Rd                         |    2
 stochtree-0.4.0/stochtree/man/JSONCreation.Rd                              |only
 stochtree-0.4.0/stochtree/man/Outcome.Rd                                   |    6
 stochtree-0.4.0/stochtree/man/OutcomeModel.Rd                              |only
 stochtree-0.4.0/stochtree/man/RandomEffectSamples.Rd                       |   44
 stochtree-0.4.0/stochtree/man/RandomEffectSamplesSerialization.Rd          |only
 stochtree-0.4.0/stochtree/man/RandomEffectStateManagement.Rd               |only
 stochtree-0.4.0/stochtree/man/RandomEffectsDataset.Rd                      |    2
 stochtree-0.4.0/stochtree/man/RandomEffectsModel.Rd                        |    2
 stochtree-0.4.0/stochtree/man/RandomEffectsTracker.Rd                      |    2
 stochtree-0.4.0/stochtree/man/bart.Rd                                      |    5
 stochtree-0.4.0/stochtree/man/bcf.Rd                                       |    5
 stochtree-0.4.0/stochtree/man/calibrateInverseGammaErrorVariance.Rd        |    6
 stochtree-0.4.0/stochtree/man/computeBARTPosteriorInterval.Rd              |only
 stochtree-0.4.0/stochtree/man/computeBCFPosteriorInterval.Rd               |only
 stochtree-0.4.0/stochtree/man/computeContrastBARTModel.Rd                  |only
 stochtree-0.4.0/stochtree/man/computeContrastBCFModel.Rd                   |only
 stochtree-0.4.0/stochtree/man/createCppRNG.Rd                              |    2
 stochtree-0.4.0/stochtree/man/createForest.Rd                              |    4
 stochtree-0.4.0/stochtree/man/createForestDataset.Rd                       |    4
 stochtree-0.4.0/stochtree/man/createForestModel.Rd                         |    4
 stochtree-0.4.0/stochtree/man/createForestModelConfig.Rd                   |   10
 stochtree-0.4.0/stochtree/man/createForestSamples.Rd                       |    4
 stochtree-0.4.0/stochtree/man/createGlobalModelConfig.Rd                   |    4
 stochtree-0.4.0/stochtree/man/createOutcome.Rd                             |    4
 stochtree-0.4.0/stochtree/man/createRandomEffectSamples.Rd                 |    4
 stochtree-0.4.0/stochtree/man/createRandomEffectsDataset.Rd                |    4
 stochtree-0.4.0/stochtree/man/createRandomEffectsModel.Rd                  |    4
 stochtree-0.4.0/stochtree/man/createRandomEffectsTracker.Rd                |    4
 stochtree-0.4.0/stochtree/man/extractParameter.Rd                          |only
 stochtree-0.4.0/stochtree/man/extractParameter.bartmodel.Rd                |only
 stochtree-0.4.0/stochtree/man/extractParameter.bcfmodel.Rd                 |only
 stochtree-0.4.0/stochtree/man/getRandomEffectSamples.Rd                    |    2
 stochtree-0.4.0/stochtree/man/getRandomEffectSamples.bartmodel.Rd          |    2
 stochtree-0.4.0/stochtree/man/getRandomEffectSamples.bcfmodel.Rd           |    2
 stochtree-0.4.0/stochtree/man/loadScalarJson.Rd                            |    4
 stochtree-0.4.0/stochtree/man/loadVectorJson.Rd                            |    4
 stochtree-0.4.0/stochtree/man/plot.bartmodel.Rd                            |    2
 stochtree-0.4.0/stochtree/man/plot.bcfmodel.Rd                             |    2
 stochtree-0.4.0/stochtree/man/predict.bartmodel.Rd                         |    4
 stochtree-0.4.0/stochtree/man/predict.bcfmodel.Rd                          |    2
 stochtree-0.4.0/stochtree/man/print.ForestSamples.Rd                       |only
 stochtree-0.4.0/stochtree/man/print.RandomEffectSamples.Rd                 |only
 stochtree-0.4.0/stochtree/man/print.bartmodel.Rd                           |    2
 stochtree-0.4.0/stochtree/man/print.bcfmodel.Rd                            |    2
 stochtree-0.4.0/stochtree/man/sampleBARTPosteriorPredictive.Rd             |only
 stochtree-0.4.0/stochtree/man/sampleBCFPosteriorPredictive.Rd              |only
 stochtree-0.4.0/stochtree/man/sampleGlobalErrorVarianceOneIteration.Rd     |    4
 stochtree-0.4.0/stochtree/man/sampleLeafVarianceOneIteration.Rd            |    4
 stochtree-0.4.0/stochtree/man/sample_without_replacement.Rd                |    4
 stochtree-0.4.0/stochtree/man/stochtree-package.Rd                         |    3
 stochtree-0.4.0/stochtree/man/summary.bartmodel.Rd                         |    2
 stochtree-0.4.0/stochtree/man/summary.bcfmodel.Rd                          |    2
 stochtree-0.4.0/stochtree/src/Makevars.in                                  |    1
 stochtree-0.4.0/stochtree/src/Makevars.win.in                              |    1
 stochtree-0.4.0/stochtree/src/R_data.cpp                                   |   31
 stochtree-0.4.0/stochtree/src/cpp11.cpp                                    |   77
 stochtree-0.4.0/stochtree/src/forest.cpp                                   |    6
 stochtree-0.4.0/stochtree/src/include/stochtree/data.h                     |   33
 stochtree-0.4.0/stochtree/src/include/stochtree/leaf_model.h               |  636 +++--
 stochtree-0.4.0/stochtree/src/include/stochtree/ordinal_sampler.h          |only
 stochtree-0.4.0/stochtree/src/include/stochtree/partition_tracker.h        |    2
 stochtree-0.4.0/stochtree/src/include/stochtree/tree_sampler.h             |   51
 stochtree-0.4.0/stochtree/src/kernel.cpp                                   |    1
 stochtree-0.4.0/stochtree/src/leaf_model.cpp                               |   60
 stochtree-0.4.0/stochtree/src/ordinal_sampler.cpp                          |only
 stochtree-0.4.0/stochtree/src/partition_tracker.cpp                        |   56
 stochtree-0.4.0/stochtree/src/sampler.cpp                                  |  150 -
 stochtree-0.4.0/stochtree/src/stochtree_types.h                            |    2
 stochtree-0.4.0/stochtree/tests/testthat/test-api-combinations.R           |   12
 stochtree-0.4.0/stochtree/tests/testthat/test-bart.R                       |  512 ++++
 stochtree-0.4.0/stochtree/tests/testthat/test-extract-parameter.R          |only
 stochtree-0.4.0/stochtree/tests/testthat/test-forest-container.R           |   24
 stochtree-0.4.0/stochtree/tests/testthat/test-predict.R                    | 1051 ++++++++
 stochtree-0.4.0/stochtree/tests/testthat/test-print-summary.R              |only
 stochtree-0.4.0/stochtree/tests/testthat/test-random-effects.R             |  101
 stochtree-0.4.0/stochtree/tests/testthat/test-utils.R                      |   32
 160 files changed, 5195 insertions(+), 2227 deletions(-)

More information about stochtree at CRAN