getRandomEffectSamples {stochtree} | R Documentation |
Generic function for extracting random effect samples from a model object (BCF, BART, etc...)
Description
Generic function for extracting random effect samples from a model object (BCF, BART, etc...)
Usage
getRandomEffectSamples(object, ...)
Arguments
object |
Fitted model object from which to extract random effects |
... |
Other parameters to be used in random effects extraction |
Value
List of random effect samples
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)
[Package stochtree version 0.1.1 Index]