File tree Expand file tree Collapse file tree 1 file changed +32
-0
lines changed
Expand file tree Collapse file tree 1 file changed +32
-0
lines changed Original file line number Diff line number Diff line change 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+ }
You can’t perform that action at this time.
0 commit comments