Extract Random Effects Samples from BART Model (legacy alias) — getRandomEffectSamples.bartmodel • stochtree

Legacy alias for extractRandomEffectSamples(); delegates to it.

Usage

# S3 method for class 'bartmodel'
getRandomEffectSamples(object, ...)

Arguments

object: Object of type bartmodel containing draws of a BART model and associated sampling outputs.
...: Other parameters to be used in random effects extraction

Value

List of random effect samples (see extractRandomEffectSamples()).

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)
rfx_samples <- getRandomEffectSamples(bart_model)