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

Documentation for package ‘stochtree’ version 0.4.0

Help Pages

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