Skip to content

Commit 089759a

Browse files
author
Bob Carpenter
committed
finished manual for next version other than possible addition of fake data and posterior predictive chapter
1 parent 054c358 commit 089759a

File tree

1 file changed

+32
-0
lines changed

1 file changed

+32
-0
lines changed
Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
data {
2+
int<lower=0> N; // num individuals
3+
int<lower=1> K; // num ind predictors
4+
int<lower=1> J; // num groups
5+
int<lower=1> L; // num group predictors
6+
int<lower=1,upper=J> jj[N]; // group for individual
7+
matrix[N,K] x; // individual predictors
8+
matrix[J,L] u; // group predictors
9+
vector[N] y; // outcomes
10+
}
11+
parameters {
12+
corr_matrix[K] Omega; // prior covariance
13+
vector<lower=0>[K] tau; // prior scale
14+
matrix[L,K] gamma; // group coeffs
15+
vector[K] beta[J]; // indiv coeffs by group
16+
real<lower=0> sigma; // prediction error scale
17+
}
18+
model {
19+
matrix[K,K] Sigma_beta;
20+
Sigma_beta <- diag_matrix(tau) * Omega * diag_matrix(tau);
21+
22+
tau ~ cauchy(0,2.5);
23+
Omega ~ lkj_corr(2);
24+
for (l in 1:L)
25+
gamma[l] ~ normal(0,5);
26+
27+
for (j in 1:J)
28+
beta[j] ~ multi_normal((u[j] * gamma)', Sigma_beta);
29+
30+
for (n in 1:N)
31+
y[n] ~ normal(x[n] * beta[jj[n]], sigma);
32+
}

0 commit comments

Comments
 (0)