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


[Up] [Top]

Documentation for package ‘stochtree’ version 0.3.0

Help Pages

stochtree-package stochtree: Stochastic Tree Ensembles (XBART and BART) for Supervised Learning and Causal Inference
bart Run BART for Supervised Learning
bcf Run BCF for Causal Effect Estimation
calibrateInverseGammaErrorVariance Calibrate Inverse Gamma Prior
computeForestLeafIndices Query Forest Leaf Indices
computeForestLeafVariances Query Forest Leaf Scale Parameters
computeForestMaxLeafIndex Query Forest Max Leaf Index
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
compute_contrast_bcf_model Compute Contrast for BCF Model
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
createCppJsonFile Create CppJson Object from File
createCppJsonString Create CppJson Object from String
createCppRNG Create CppRNG Object
createForest Create Forest 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
createPreprocessorFromJson Reload Covariate Preprocessor from JSON String
createPreprocessorFromJsonString Reload Covariate Preprocessor from JSON String
createRandomEffectSamples Create RandomEffectSamples Object
createRandomEffectsDataset Create RandomEffectsDataset Object
createRandomEffectsModel Create RandomEffectsModel Object
createRandomEffectsTracker Create RandomEffectsTracker Object
extract_parameter Extract Parameter Samples Generic Function
extract_parameter.bartmodel Extract BART Parameter Samples.
extract_parameter.bcfmodel Extract BCF Parameter Samples
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 Extract Random Effect Samples Generic Function
getRandomEffectSamples.bartmodel Extract Random Effects Samples
getRandomEffectSamples.bcfmodel Extract Random Effect Samples from BCF Model
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
loadVectorJson Load Vector from JSON
Outcome Outcome Data C++ Wrapper
plot.bartmodel Plot BART Model Fit.
plot.bcfmodel Plot BCF Model
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
sampleGlobalErrorVarianceOneIteration Sample Global Error Variance
sampleLeafVarianceOneIteration Sample Leaf Scale
sample_bart_posterior_predictive Sample BART Posterior Predictive
sample_bcf_posterior_predictive Sample BCF Posterior Predictive
sample_without_replacement Sample Without Replacement
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
saveBCFModelToJsonFile Save BCF Model to JSON File
saveBCFModelToJsonString Convert BCF Model to JSON String
savePreprocessorToJsonString Convert Covariate Preprocessor to JSON String
stochtree stochtree: Stochastic Tree Ensembles (XBART and BART) for Supervised Learning and Causal Inference
summary.bartmodel Summarize the BART model fit and sampled terms.
summary.bcfmodel Summarize BCF Model