stappDP_merdecomp: Penalized Functional Dirichlet Process Linear Mixed Effects...

In apeterson91/rstapDP: Functional Dirichlet Process Spatial Temporal Aggregated Predictors in R

Penalized Functional Dirichlet Process Linear Mixed Effects Regression with Between-Within Decomposition

Description

fits a functional dirichlet process linear mixed effects regression model with N observations and n subjects using a between-within effects decomposition on the STAP-DP design matrix

Usage

stappDP_merdecomp(
  y,
  Z,
  X_b,
  X_w,
  W,
  w,
  subj_mat_,
  subj_n,
  alpha_a,
  alpha_b,
  sigma_a,
  sigma_b,
  tau_a,
  tau_b,
  K,
  subset_one,
  subset_two,
  threshold,
  iter_max,
  burn_in,
  thin,
  seed,
  num_posterior_samples,
  chain,
  fix_alpha,
  logging
)

Arguments

y	a vector of continuous outcomes
Z	a matrix of population level confounders
X_b	Matrix of between subject spatial temporal aggregated predictor covariates
X_w	Matrix of within subject spatial temporal aggregated predictor covariates
W	a design matrix for group specific terms
w	a vector of weights for weighted regression
subj_mat_	N x n sparse matrix used to aggregate subject observations
subj_n	n x 1 vector of integers representing how many observations correspond to each subject
alpha_a	alpha gamma prior shape hyperparameter
alpha_b	alpha gamma prior scale hyperparameter
sigma_a	precision gamma prior shape hyperparameter
sigma_b	precision gamma prior scale hyperparameter
tau_a	penalty gamma prior shape hyperparameter
tau_b	penalty gamma prior scale hyperparameter
K	truncation number
subset_one	rank of first smoothing matrix
subset_two	rank of second smoothing matrix
threshold	number of members per cluster at which cluster is included in regression
iter_max	maximum number of iterations
burn_in	number of burn in iterations
thin	number by which to thin samples
seed	rng initializer
num_posterior_samples	total number of posterior samples
chain	chain label
fix_alpha	boolean value that determines whether or not to fix alpha in sampler
logging	boolean parameter indicating whether or not a single iteration should be run with print messages indicating successful completion of the Sampler's sub modules

apeterson91/rstapDP documentation built on Sept. 20, 2023, 9:34 a.m.