Skip to main content

Browse the archive

https://github.com/stan-dev/example-models
24 April 2024, 12:10:14 UTC
Revision 945e4c69971ed73ded8668418fe9c286f90c89cb authored by jrnold on 14 August 2016, 21:04:28 UTC, committed by jrnold on 14 August 2016, 21:04:28 UTC 
Remove incomplete Nile example from misc/dlm
1 parent ec6d329
Raw File
Tip revision: 945e4c69971ed73ded8668418fe9c286f90c89cb authored by jrnold on 14 August 2016, 21:04:28 UTC
Remove incomplete Nile example from misc/dlm
Tip revision: 945e4c6
hiv_inter_chr.stan 
data {
  int<lower=0> J; 
  int<lower=0> N;
  int<lower=1,upper=J> person[N];
  vector[N] time;
  vector[N] treatment;
  vector[N] y;
} 
parameters {
  real beta;
  vector[J] eta1;
  vector[J] eta2;
  real mu_a1;
  real mu_a2;
  real<lower=0,upper=100> sigma_a1;
  real<lower=0,upper=100> sigma_a2;
  real<lower=0,upper=100> sigma_y;
}
transformed parameters {
  vector[J] a1;
  vector[J] a2;
  vector[N] y_hat;

  a1 <- 10 * mu_a1 + sigma_a1 * eta1;
  a2 <- 0.1 * mu_a2 + sigma_a2 * eta2;

  for (i in 1:N)
    y_hat[i] <- beta * time[i] * treatment[i] + a1[person[i]] 
                + a2[person[i]] * time[i];
} 
model {
  mu_a1 ~ normal(0, 1);
  eta1 ~ normal (0, 1);
  mu_a2 ~ normal(0, 1);
  eta2 ~ normal (0, 1);

  beta ~ normal (0, 1);

  y ~ normal(y_hat, sigma_y);
}
back to top