Man pages for stochtree
Stochastic Tree Ensembles (XBART and BART) for Supervised Learning and Causal Inference
| bart | Run the BART algorithm for supervised learning. |
| bcf | Run the Bayesian Causal Forest (BCF) algorithm for... |
| calibrateInverseGammaErrorVariance | Calibrate the scale parameter on an inverse gamma prior for... |
| computeForestLeafIndices | Compute vector of forest leaf indices |
| computeForestLeafVariances | Compute vector of forest leaf scale parameters |
| computeForestMaxLeafIndex | Compute and return the largest possible leaf index computable... |
| convertPreprocessorToJson | Convert the persistent aspects of a covariate preprocessor to... |
| CppJson | Class that stores draws from an random ensemble of decision... |
| CppRNG | Class that wraps a C++ random number generator (for... |
| createBARTModelFromCombinedJson | Convert a list of (in-memory) JSON representations of a BART... |
| createBARTModelFromCombinedJsonString | Convert a list of (in-memory) JSON strings that represent... |
| createBARTModelFromJson | Convert an (in-memory) JSON representation of a BART model to... |
| createBARTModelFromJsonFile | Convert a JSON file containing sample information on a... |
| createBARTModelFromJsonString | Convert a JSON string containing sample information on a... |
| createBCFModelFromCombinedJson | Convert a list of (in-memory) JSON strings that represent BCF... |
| createBCFModelFromCombinedJsonString | Convert a list of (in-memory) JSON strings that represent BCF... |
| createBCFModelFromJson | Convert an (in-memory) JSON representation of a BCF model to... |
| createBCFModelFromJsonFile | Convert a JSON file containing sample information on a... |
| createBCFModelFromJsonString | Convert a JSON string containing sample information on a... |
| createCppJson | Create a new (empty) C++ Json object |
| createCppJsonFile | Create a C++ Json object from a Json file |
| createCppJsonString | Create a C++ Json object from a Json string |
| createCppRNG | Create an R class that wraps a C++ random number generator |
| createForest | Create a forest |
| createForestDataset | Create a forest dataset object |
| createForestModel | Create a forest model object |
| createForestModelConfig | Create a forest model config object |
| createForestSamples | Create a container of forest samples |
| createGlobalModelConfig | Create a global model config object |
| createOutcome | Create an outcome object |
| createPreprocessorFromJson | Reload a covariate preprocessor object from a JSON string... |
| createPreprocessorFromJsonString | Reload a covariate preprocessor object from a JSON string... |
| createRandomEffectSamples | Create a 'RandomEffectSamples' object |
| createRandomEffectsDataset | Create a random effects dataset object |
| createRandomEffectsModel | Create a 'RandomEffectsModel' object |
| createRandomEffectsTracker | Create a 'RandomEffectsTracker' object |
| Forest | Class that stores a single ensemble of decision trees (often... |
| ForestDataset | Dataset used to sample a forest |
| ForestModel | Class that defines and samples a forest model |
| ForestModelConfig | Object used to get / set parameters and other model... |
| ForestSamples | Class that stores draws from an random ensemble of decision... |
| getRandomEffectSamples | Generic function for extracting random effect samples from a... |
| getRandomEffectSamples.bartmodel | Extract raw sample values for each of the random effect... |
| getRandomEffectSamples.bcfmodel | Extract raw sample values for each of the random effect... |
| GlobalModelConfig | Object used to get / set global parameters and other global... |
| loadForestContainerCombinedJson | Combine multiple JSON model objects containing forests (with... |
| loadForestContainerCombinedJsonString | Combine multiple JSON strings representing model objects... |
| loadForestContainerJson | Load a container of forest samples from json |
| loadRandomEffectSamplesCombinedJson | Combine multiple JSON model objects containing random effects... |
| loadRandomEffectSamplesCombinedJsonString | Combine multiple JSON strings representing model objects... |
| loadRandomEffectSamplesJson | Load a container of random effect samples from json |
| loadScalarJson | Load a scalar from json |
| loadVectorJson | Load a vector from json |
| Outcome | Outcome / partial residual used to sample an additive model. |
| predict.bartmodel | Predict from a sampled BART model on new data |
| predict.bcfmodel | Predict from a sampled BCF model on new data |
| preprocessPredictionData | Preprocess covariates. DataFrames will be preprocessed based... |
| preprocessTrainData | Preprocess covariates. DataFrames will be preprocessed based... |
| RandomEffectSamples | Class that wraps the "persistent" aspects of a C++ random... |
| RandomEffectsDataset | Dataset used to sample a random effects model |
| RandomEffectsModel | The core "model" class for sampling random effects. |
| RandomEffectsTracker | Class that defines a "tracker" for random effects models,... |
| resetActiveForest | Reset an active forest, either from a specific forest in a... |
| resetForestModel | Re-initialize a forest model (tracking data structures) from... |
| resetRandomEffectsModel | Reset a 'RandomEffectsModel' object based on the parameters... |
| resetRandomEffectsTracker | Reset a 'RandomEffectsTracker' object based on the parameters... |
| rootResetRandomEffectsModel | Reset a 'RandomEffectsModel' object to its "default" state |
| rootResetRandomEffectsTracker | Reset a 'RandomEffectsTracker' object to its "default" state |
| sampleGlobalErrorVarianceOneIteration | Sample one iteration of the (inverse gamma) global variance... |
| sampleLeafVarianceOneIteration | Sample one iteration of the leaf parameter variance model... |
| saveBARTModelToJson | Convert the persistent aspects of a BART model to (in-memory)... |
| saveBARTModelToJsonFile | Convert the persistent aspects of a BART model to (in-memory)... |
| saveBARTModelToJsonString | Convert the persistent aspects of a BART model to (in-memory)... |
| saveBCFModelToJson | Convert the persistent aspects of a BCF model to (in-memory)... |
| saveBCFModelToJsonFile | Convert the persistent aspects of a BCF model to (in-memory)... |
| saveBCFModelToJsonString | Convert the persistent aspects of a BCF model to (in-memory)... |
| savePreprocessorToJsonString | Convert the persistent aspects of a covariate preprocessor to... |
| stochtree-package | stochtree: Stochastic Tree Ensembles (XBART and BART) for... |
