Skip to content

Commit 19b682b

Browse files
committed
bugs example v2/ice uses poisson+log
1 parent b203aea commit 19b682b

File tree

1 file changed

+7
-12
lines changed

1 file changed

+7
-12
lines changed

bugs_examples/vol2/ice/ice.stan

Lines changed: 7 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -28,24 +28,19 @@ parameters {
2828
}
2929

3030
model {
31-
real yaalpha[Nage];
31+
vector[N] r;
3232

3333
sigma ~ uniform(0, 1);
3434
for (k in 1:2) beta[k] ~ normal(0, sigma * 1E3);
3535
for (k in 3:K) beta[k] ~ normal(2 * beta[k - 1] - beta[k - 2], sigma);
36+
alpha ~ normal(0, 1000);
3637

37-
// for (k in 1:K)
38-
// logRR[k] <- beta[k] - beta[5];
38+
for (i in 1:N) {
39+
if (age[i] == 1) r[i] <- alpha1 + log(pyr[i]) + beta[year[i]];
40+
else r[i] <- alpha[age[i] - 1] + log(pyr[i]) + beta[year[i]];
41+
}
3942

40-
for (j in 2:Nage) alpha[j - 1] ~ normal(0, 1000);
41-
42-
// real logRR[K];
43-
yaalpha[1] <- alpha1;
44-
for (i in 2:Nage)
45-
yaalpha[i] <- alpha[i - 1];
46-
47-
for (i in 1:N)
48-
cases[i] ~ poisson(exp(log(pyr[i]) + yaalpha[age[i]] + beta[year[i]]));
43+
cases ~ poisson_log(r);
4944
}
5045

5146
generated quantities {

0 commit comments

Comments
 (0)