fits a functional dirichlet process linear mixed effects regression model with N observations and n subjects

stappDP_mer_fit(
  y,
  Z,
  X,
  W,
  S,
  w,
  subj_mat_,
  subj_n,
  alpha_a,
  alpha_b,
  sigma_a,
  sigma_b,
  tau_a,
  tau_b,
  K,
  num_penalties,
  iter_max,
  burn_in,
  thin,
  seed,
  num_posterior_samples,
  fix_alpha
)

Arguments

y

a vector of continuous outcomes

Z

a matrix of population level confounders

X

a matrix of spatial temporal aggregated predictors

W

a design matrix for group specific terms

S

penalty matrix for stap parameters

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

num_penalties

number of penalty matrices accounted for in S

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

fix_alpha

boolean value that determines whether or not to fix alpha in sampler