sesame_multi_preds_3a.stan

data {
  int<lower=0> N;
  vector[N] encouraged;
  vector[N] setting;
  array[N] int site;
  vector[N] pretest;
  vector[N] watched;
}
transformed data {
  vector[N] site2;
  vector[N] site3;
  vector[N] site4;
  vector[N] site5;
  matrix[N, 7] x;
  
  for (i in 1 : N) {
    site2[i] = site[i] == 2;
    site3[i] = site[i] == 3;
    site4[i] = site[i] == 4;
    site5[i] = site[i] == 5;
  }
  
  x = [encouraged', pretest', site2', site3', site4', site5', setting']';
}
parameters {
  real alpha;
  vector[7] beta;
  real<lower=0> sigma;
}
model {
  watched ~ normal_id_glm(x, alpha, beta, sigma);
}