createRandomEffectsDataset: Create a random effects dataset object

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

Create a random effects dataset object

Description

Create a random effects dataset object

Usage

createRandomEffectsDataset(group_labels, basis, variance_weights = NULL)

Arguments

group_labels	Vector of group labels
basis	Matrix of bases used to define the random effects regression (for an intercept-only model, pass an array of ones)
variance_weights	(Optional) Vector of observation-specific variance weights

Value

RandomEffectsDataset object

Examples

rfx_group_ids <- sample(1:2, size = 100, replace = TRUE)
rfx_basis <- matrix(rnorm(3*100), ncol = 3)
weight_vector <- rnorm(100)
rfx_dataset <- createRandomEffectsDataset(rfx_group_ids, rfx_basis)
rfx_dataset <- createRandomEffectsDataset(rfx_group_ids, rfx_basis, weight_vector)

stochtree documentation built on April 4, 2025, 2:11 a.m.