Skip to content

Commit a42b3da

Browse files
WardBrianWardBrian
committed
Run canonicalizer on all models
2 parents 885bd18 + 8c85b1c commit a42b3da

File tree

531 files changed

+10710
-10410
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

531 files changed

+10710
-10410
lines changed

‎ARM/Ch.10/ideo_interactions.stan

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,18 @@
11
data {
2-
int<lower=0> N;
2+
int<lower=0> N;
33
vector[N] party;
44
vector[N] score1;
55
vector[N] x;
66
}
77
transformed data {
88
vector[N] inter = party .* x;
9-
matrix[N,3] cov = [party', x', inter']';
9+
matrix[N, 3] cov = [party', x', inter']';
1010
}
1111
parameters {
1212
real alpha;
1313
vector[3] beta;
1414
real<lower=0> sigma;
15-
}
15+
}
1616
model {
1717
score1 ~ normal_id_glm(cov, alpha, beta, sigma);
1818
}

‎ARM/Ch.10/ideo_reparam.stan

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,18 @@
11
data {
2-
int<lower=0> N;
2+
int<lower=0> N;
33
vector[N] party;
44
vector[N] score1;
55
vector[N] z1; //z value for party 0, 0 otherwise
66
vector[N] z2; //z value for party 1, 0 otherwise
77
}
88
transformed data {
9-
matrix[N,3] x = [party', z1', z2']';
9+
matrix[N, 3] x = [party', z1', z2']';
1010
}
1111
parameters {
1212
real alpha;
1313
vector[3] beta;
1414
real<lower=0> sigma;
15-
}
15+
}
1616
model {
1717
score1 ~ normal_id_glm(x, alpha, beta, sigma);
1818
}

‎ARM/Ch.10/ideo_two_pred.stan

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,17 @@
11
data {
2-
int<lower=0> N;
2+
int<lower=0> N;
33
vector[N] party;
44
vector[N] score1;
55
vector[N] x;
66
}
77
transformed data {
8-
matrix[N,2] cov = [party', x']';
8+
matrix[N, 2] cov = [party', x']';
99
}
1010
parameters {
1111
real alpha;
1212
vector[2] beta;
1313
real<lower=0> sigma;
14-
}
14+
}
1515
model {
1616
score1 ~ normal_id_glm(cov, alpha, beta, sigma);
1717
}

‎ARM/Ch.10/sesame_multi_preds_3a.stan

+11-12
Original file line numberDiff line numberDiff line change
@@ -1,33 +1,32 @@
11
data {
2-
int<lower=0> N;
2+
int<lower=0> N;
33
vector[N] encouraged;
44
vector[N] setting;
5-
int site[N];
5+
array[N] int site;
66
vector[N] pretest;
77
vector[N] watched;
88
}
99
transformed data {
10-
vector[N] site2;
11-
vector[N] site3;
12-
vector[N] site4;
10+
vector[N] site2;
11+
vector[N] site3;
12+
vector[N] site4;
1313
vector[N] site5;
14-
matrix[N,7] x;
15-
16-
for (i in 1:N) {
14+
matrix[N, 7] x;
15+
16+
for (i in 1 : N) {
1717
site2[i] = site[i] == 2;
1818
site3[i] = site[i] == 3;
1919
site4[i] = site[i] == 4;
2020
site5[i] = site[i] == 5;
2121
}
22-
23-
x = [encouraged', pretest', site2', site3',
24-
site4', site5', setting']';
22+
23+
x = [encouraged', pretest', site2', site3', site4', site5', setting']';
2524
}
2625
parameters {
2726
real alpha;
2827
vector[7] beta;
2928
real<lower=0> sigma;
30-
}
29+
}
3130
model {
3231
watched ~ normal_id_glm(x, alpha, beta, sigma);
3332
}

‎ARM/Ch.10/sesame_multi_preds_3b.stan

+11-12
Original file line numberDiff line numberDiff line change
@@ -1,33 +1,32 @@
11
data {
2-
int<lower=0> N;
2+
int<lower=0> N;
33
vector[N] pretest;
44
vector[N] setting;
5-
int site[N];
5+
array[N] int site;
66
vector[N] watched_hat;
77
vector[N] y;
88
}
99
transformed data {
10-
vector[N] site2;
11-
vector[N] site3;
12-
vector[N] site4;
10+
vector[N] site2;
11+
vector[N] site3;
12+
vector[N] site4;
1313
vector[N] site5;
14-
matrix[N,7] x;
15-
16-
for (i in 1:N) {
14+
matrix[N, 7] x;
15+
16+
for (i in 1 : N) {
1717
site2[i] = site[i] == 2;
1818
site3[i] = site[i] == 3;
1919
site4[i] = site[i] == 4;
2020
site5[i] = site[i] == 5;
2121
}
22-
23-
x = [watched_hat', pretest', site2', site3',
24-
site4', site5', setting']';
22+
23+
x = [watched_hat', pretest', site2', site3', site4', site5', setting']';
2524
}
2625
parameters {
2726
real alpha;
2827
vector[7] beta;
2928
real<lower=0> sigma;
30-
}
29+
}
3130
model {
3231
y ~ normal_id_glm(x, alpha, beta, sigma);
3332
}

‎ARM/Ch.10/sesame_one_pred_2b.stan

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,16 @@
11
data {
2-
int<lower=0> N;
2+
int<lower=0> N;
33
vector[N] watched_hat;
44
vector[N] y;
55
}
66
transformed data {
7-
matrix[N,1] x = [watched_hat']';
7+
matrix[N, 1] x = [watched_hat']';
88
}
99
parameters {
1010
real alpha;
1111
vector[1] beta;
1212
real<lower=0> sigma;
13-
}
13+
}
1414
model {
1515
y ~ normal_id_glm(x, alpha, beta, sigma);
1616
}

‎ARM/Ch.10/sesame_one_pred_a.stan

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,16 @@
11
data {
2-
int<lower=0> N;
2+
int<lower=0> N;
33
vector[N] encouraged;
44
vector[N] watched;
55
}
66
transformed data {
7-
matrix[N,1] x = [encouraged']';
7+
matrix[N, 1] x = [encouraged']';
88
}
99
parameters {
1010
real alpha;
1111
vector[1] beta;
1212
real<lower=0> sigma;
13-
}
13+
}
1414
model {
1515
watched ~ normal_id_glm(x, alpha, beta, sigma);
1616
}

‎ARM/Ch.10/sesame_one_pred_b.stan

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,16 @@
11
data {
2-
int<lower=0> N;
2+
int<lower=0> N;
33
vector[N] encouraged;
44
vector[N] y;
55
}
66
transformed data {
7-
matrix[N,1] x = [encouraged']';
7+
matrix[N, 1] x = [encouraged']';
88
}
99
parameters {
1010
real alpha;
1111
vector[1] beta;
1212
real<lower=0> sigma;
13-
}
13+
}
1414
model {
1515
y ~ normal_id_glm(x, alpha, beta, sigma);
1616
}

‎ARM/Ch.12/radon_complete_pool.stan

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ data {
44
vector[N] y;
55
}
66
transformed data {
7-
matrix[N,1] cov = [x']';
7+
matrix[N, 1] cov = [x']';
88
}
99
parameters {
1010
real alpha;

‎ARM/Ch.12/radon_group.stan

+4-4
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
data {
22
int<lower=1> N;
3-
int<lower=1> J; # number of counties
4-
int<lower=1,upper=J> county[N];
3+
int<lower=1> J; // number of counties
4+
array[N] int<lower=1, upper=J> county;
55
vector[N] u;
66
vector[N] x;
77
vector[N] y;
88
}
99
transformed data {
10-
matrix[N,2] cov = [x', u']';
10+
matrix[N, 2] cov = [x', u']';
1111
}
1212
parameters {
1313
real mu_alpha;
@@ -26,6 +26,6 @@ model {
2626
sigma_alpha ~ cauchy(0, 2.5);
2727
mu_beta ~ std_normal();
2828
sigma_beta ~ cauchy(0, 2.5);
29-
29+
3030
y ~ normal_id_glm(cov, alpha[county], beta, sigma);
3131
}

‎ARM/Ch.12/radon_group_chr.stan

+4-4
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
data {
22
int<lower=1> N;
3-
int<lower=1> J; # number of counties
4-
int<lower=1,upper=J> county[N];
3+
int<lower=1> J; // number of counties
4+
array[N] int<lower=1, upper=J> county;
55
vector[N] u;
66
vector[N] x;
77
vector[N] y;
88
}
99
transformed data {
10-
matrix[N,2] cov = [x', u']';
10+
matrix[N, 2] cov = [x', u']';
1111
}
1212
parameters {
1313
vector[2] beta;
@@ -22,6 +22,6 @@ model {
2222
sigma ~ cauchy(0, 2.5);
2323
sigma_a ~ cauchy(0, 2.5);
2424
alpha ~ normal(mu_a, sigma_a);
25-
25+
2626
y ~ normal_id_glm(cov, alpha[county], beta, sigma);
2727
}

‎ARM/Ch.12/radon_intercept.stan

+3-4
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,14 @@
11
data {
22
int<lower=1> N;
3-
int<lower=1> J; # number of counties
4-
int<lower=1,upper=J> county[N];
3+
int<lower=1> J; // number of counties
4+
array[N] int<lower=1, upper=J> county;
55
vector[N] y;
66
}
77
parameters {
88
real mu_a;
99
real<lower=0> sigma_a;
1010
real<lower=0> sigma_y;
11-
vector<offset=mu_a, multiplier=sigma_a>[J] a; # county intercepts
11+
vector<offset=mu_a, multiplier=sigma_a>[J] a; // county intercepts
1212
}
1313
model {
1414
mu_a ~ std_normal();
@@ -17,4 +17,3 @@ model {
1717
a ~ normal(mu_a, sigma_a);
1818
y ~ normal(a[county], sigma_y);
1919
}
20-

‎ARM/Ch.12/radon_intercept_chr.stan

+3-4
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
data {
22
int<lower=1> N;
3-
int<lower=1> J; # number of counties
4-
int<lower=1,upper=J> county[N];
3+
int<lower=1> J; // number of counties
4+
array[N] int<lower=1, upper=J> county;
55
vector[N] y;
66
}
77
parameters {
@@ -10,12 +10,11 @@ parameters {
1010
real<lower=0> sigma_y;
1111
vector<offset=(10 * mu_a), multiplier=sigma_a>[J] alpha;
1212
}
13-
1413
model {
1514
mu_a ~ std_normal();
1615
alpha ~ normal(10 * mu_a, sigma_a);
1716
sigma_a ~ cauchy(0, 2.5);
1817
sigma_y ~ cauchy(0, 2.5);
19-
18+
2019
y ~ normal(alpha[county], sigma_y);
2120
}

‎ARM/Ch.12/radon_no_pool.stan

+4-4
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
data {
22
int<lower=1> N;
3-
int<lower=1> J; # number of counties
4-
int<lower=1,upper=J> county[N];
3+
int<lower=1> J; // number of counties
4+
array[N] int<lower=1, upper=J> county;
55
vector[N] x;
66
vector[N] y;
77
}
88
transformed data {
9-
matrix[N,1] cov = [x']';
9+
matrix[N, 1] cov = [x']';
1010
}
1111
parameters {
1212
vector[1] beta;
@@ -20,7 +20,7 @@ model {
2020
mu_a ~ std_normal();
2121
sigma_a ~ cauchy(0, 2.5);
2222
sigma_y ~ cauchy(0, 2.5);
23-
23+
2424
a ~ normal(mu_a, sigma_a);
2525
y ~ normal_id_glm(cov, a[county], beta, sigma_y);
2626
}

‎ARM/Ch.12/radon_no_pool_chr.stan

+4-4
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
data {
22
int<lower=1> N;
3-
int<lower=1> J; # number of counties
4-
int<lower=1,upper=J> county[N];
3+
int<lower=1> J; // number of counties
4+
array[N] int<lower=1, upper=J> county;
55
vector[N] x;
66
vector[N] y;
77
}
88
transformed data {
9-
matrix[N,1] cov = [x']';
9+
matrix[N, 1] cov = [x']';
1010
}
1111
parameters {
1212
vector[1] beta;
@@ -21,6 +21,6 @@ model {
2121
alpha ~ normal(mu_a, sigma_a);
2222
sigma_a ~ cauchy(0, 2.5);
2323
sigma_y ~ cauchy(0, 2.5);
24-
24+
2525
y ~ normal_id_glm(cov, alpha[county], beta, sigma_y);
2626
}

0 commit comments

Comments
 (0)