Extract Forest Samples Generic Function

Generic function for extracting forest samples from a model object (BCF, BART, etc...)

Usage

extractForest(object, term)

Arguments

object

Fitted model object from which to extract forest samples

term

Name of the forest to extract (e.g., "mean", "variance", etc.)

Value

Forest sample array

Examples

n <- 100
p <- 10
X <- matrix(runif(n*p), ncol = p)
rfx_group_ids <- sample(1:2, size = n, replace = TRUE)
rfx_basis <- rep(1.0, n)
y <- (-5 + 10*(X[,1] > 0.5)) + (-2*(rfx_group_ids==1)+2*(rfx_group_ids==2)) + rnorm(n)
bart_model <- bart(X_train=X, y_train=y, rfx_group_ids_train=rfx_group_ids,
                   rfx_basis_train = rfx_basis, num_gfr=0, num_mcmc=10)
mean_forest <- extractForest(bart_model, "mean")