-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsx_pp_task_id.stan
69 lines (64 loc) · 1.62 KB
/
sx_pp_task_id.stan
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
//
data {
int<lower = 0> N; // number of participants
int<lower = 1, upper = 5> stage[N]; //stage variable
int<lower = 1> N_stages; // number of stages
int<lower = 1, upper = 13> task[N]; //task variable
int<lower = 1> N_tasks; // number of tasks
int<lower = 1, upper = 14> id[N]; //id variable
int<lower = 1> N_ids; // number of ids
vector [N]sx;
}
parameters {
real<lower = 0> nu;
vector[5]b_stage_raw;
vector[13]b_task_raw;
vector[14]b_id_raw;
real mu_stage;
real mu_task;
real mu_id;
real<lower = 0> sig_stage;
real<lower = 0> sig_task;
real<lower = 0> sig_id;
real<lower = 0>sig;
}
transformed parameters {
vector[N_stages] b_stage;
vector[N_tasks] b_task;
vector[N_ids] b_id;
for (i in 1:N_stages) {
b_stage[i] = mu_stage + sig_stage * b_stage_raw[i];
}
for (i in 1:N_tasks) {
b_task[i] = mu_task + sig_task * b_task_raw[i];
}
for (i in 1:N_ids) {
b_id[i] = mu_id + sig_id * b_id_raw[i];
}
}
model {
// priors
b_stage_raw ~ normal(0,1);
b_task_raw ~ normal(0,1);
b_id_raw ~ normal(0,1);
mu_stage ~ normal(0,0.5);
sig_stage ~ exponential( 1 );
mu_task ~ normal(0,0.2);
sig_task ~ exponential( 1 );
mu_id ~ normal(0,0.2);
sig_id ~ exponential( 1 );
sig ~ exponential( 1 );
nu ~ gamma( 2 ,0.1 );
//likelihood
for (i in 1:N) {
sx[i] ~ student_t(nu,b_stage[stage[i]] + b_task[task[i]]+ b_id[id[i]], sig);
}
}
generated quantities{
vector[N] log_lik;
vector[N] mu;
for ( i in 1:N ) {
mu[i] = b_stage[stage[i]] + b_task[task[i]] + b_id[id[i]];
}
for ( i in 1:N ) log_lik[i] = student_t_lpdf( sx[i] | nu,mu[i] , sig );
}