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


[Up] [Top]

Documentation for package ‘stochtree’ version 0.4.2

Help Pages

B C D E F G J L O P R S

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

-- B --

bart Run BART for Supervised Learning
BARTSerialization BART Serialization Routines
bcf Run BCF for Causal Effect Estimation
BCFSerialization BCF Serialization Routines

-- C --

calibrateInverseGammaErrorVariance Calibrate Inverse Gamma Prior
computeBARTPosteriorInterval Compute BART Posterior Credible Intervals
computeBCFPosteriorInterval Compute BCF Posterior Credible Intervals
computeContrastBARTModel Compute Contrast for BART Model
computeContrastBCFModel Compute Contrast for BCF Model
computeForestLeafIndices Forest Kernel Computation Routines
computeForestLeafVariances Forest Kernel Computation Routines
computeForestMaxLeafIndex Forest Kernel Computation Routines
CppJson JSON C++ Object Wrapper
CppRNG Random Number Generator C++ Wrapper
createBARTModelFromCombinedJson BART Serialization Routines
createBARTModelFromCombinedJsonString BART Serialization Routines
createBARTModelFromJson BART Serialization Routines
createBARTModelFromJsonFile BART Serialization Routines
createBARTModelFromJsonString BART Serialization Routines
createBCFModelFromCombinedJson BCF Serialization Routines
createBCFModelFromCombinedJsonString BCF Serialization Routines
createBCFModelFromJson BCF Serialization Routines
createBCFModelFromJsonFile BCF Serialization Routines
createBCFModelFromJsonString BCF Serialization Routines
createCppJson JSON Creation Routines
createCppJsonFile JSON Creation Routines
createCppJsonString JSON Creation Routines
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 Data Preprocessing Routines
createPreprocessorFromJsonString Data Preprocessing Routines
createRandomEffectSamples Create RandomEffectSamples Object
createRandomEffectsDataset Create RandomEffectsDataset Object
createRandomEffectsModel Create RandomEffectsModel Object
createRandomEffectsTracker Create RandomEffectsTracker Object

-- D --

DataPreprocessing Data Preprocessing Routines

-- E --

extractParameter Extract Parameter Samples Generic Function
extractParameter.bartmodel Extract BART Parameter Samples
extractParameter.bcfmodel Extract BCF Parameter Samples

-- F --

Forest Forest C++ Wrapper
ForestDataset Forest Dataset C++ Wrapper
ForestKernelComputation Forest Kernel Computation Routines
ForestModel Forest Model C++ Wrapper
ForestModelConfig Forest Model Configuration Object
ForestSamples Forest Container C++ Wrapper
ForestSamplesSerialization Forest Container Serialization Routines
ForestStateManagement Forest State Management Routines

-- G --

getRandomEffectSamples Extract Random Effect Samples Generic Function
getRandomEffectSamples.bartmodel Extract Random Effects Samples from BART Model
getRandomEffectSamples.bcfmodel Extract Random Effect Samples from BCF Model
GlobalModelConfig Global Model Configuration Object

-- J --

JSONCreation JSON Creation Routines

-- L --

loadForestContainerCombinedJson Forest Container Serialization Routines
loadForestContainerCombinedJsonString Forest Container Serialization Routines
loadForestContainerJson Forest Container Serialization Routines
loadRandomEffectSamplesCombinedJson Random Effects Container Serialization Routines
loadRandomEffectSamplesCombinedJsonString Random Effects Container Serialization Routines
loadRandomEffectSamplesJson Random Effects Container Serialization Routines
loadScalarJson Load Scalar from JSON
loadVectorJson Load Vector from JSON

-- O --

Outcome Outcome Data C++ Wrapper
OutcomeModel Create a New Outcome Model Object

-- P --

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 Data Preprocessing Routines
preprocessTrainData Data Preprocessing Routines
print.bartmodel Print Summary of BART Model
print.bcfmodel Print Summary of BCF Model
print.ForestSamples Summarize a ForestSamples Object
print.RandomEffectSamples Summarize a RandomEffectSamples Object

-- R --

RandomEffectSamples Random Effect Container C++ Wrapper
RandomEffectSamplesSerialization Random Effects Container Serialization Routines
RandomEffectsDataset Random Effects Dataset C++ Wrapper
RandomEffectsModel Random Effects Model C++ Wrapper
RandomEffectStateManagement Random Effects State Management Routines
RandomEffectsTracker Random Effects Tracker C++ Wrapper
resetActiveForest Forest State Management Routines
resetForestModel Forest State Management Routines
resetRandomEffectsModel Random Effects State Management Routines
resetRandomEffectsTracker Random Effects State Management Routines
rootResetRandomEffectsModel Random Effects State Management Routines
rootResetRandomEffectsTracker Random Effects State Management Routines

-- S --

sampleBARTPosteriorPredictive Sample BART Posterior Predictive
sampleBCFPosteriorPredictive Sample BCF Posterior Predictive
sampleGlobalErrorVarianceOneIteration Sample Global Error Variance
sampleLeafVarianceOneIteration Sample Leaf Scale
sample_without_replacement Sample Without Replacement
saveBARTModelToJson BART Serialization Routines
saveBARTModelToJsonFile BART Serialization Routines
saveBARTModelToJsonString BART Serialization Routines
saveBCFModelToJson BCF Serialization Routines
saveBCFModelToJsonFile BCF Serialization Routines
saveBCFModelToJsonString BCF Serialization Routines
savePreprocessorToJson Data Preprocessing Routines
savePreprocessorToJsonString Data Preprocessing Routines
stochtree stochtree: Stochastic Tree Ensembles (XBART and BART) for Supervised Learning and Causal Inference
summary.bartmodel Summarize BART Model Fit and Parameters
summary.bcfmodel Summarize BCF Model