Skip to content

Commit d3fc6ad

Browse files
committed
improve GCM_3_optimized model
1 parent ac43df8 commit d3fc6ad

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

Bayesian_Cognitive_Modeling/CaseStudies/GCM/!GCM_3_optimized_Stan.R

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -69,10 +69,15 @@ transformed parameters {
6969
- log(normal_cdf(1, muw[2], sigmaw)
7070
- normal_cdf(0, muw[2], sigmaw));
7171
}
72-
for (k in 1:nsubj) {
73-
vector[nstim] r;
74-
r <- exp(log_r[k]);
75-
lp_parts_c[k,1] <- log1m(phic) + binomial_log(y[k], n, r);
72+
for (k in 1:nsubj) {
73+
vector[nstim] tmp;
74+
75+
// Binomial distribution written as Log Probability Mass Function
76+
for (i in 1:nstim)
77+
tmp[i] <- lgamma(n + 1) - lgamma(y[k,i] + 1) - lgamma(n - y[k,i] + 1)
78+
+ y[k,i] * log_r[k,i] + (n - y[k,i]) * log1m_exp(log_r[k,i]);
79+
80+
lp_parts_c[k,1] <- log1m(phic) + sum(tmp);
7681
lp_parts_c[k,2] <- log(phic) + binomial_log(y[k], n, .5);
7782
}
7883
}

0 commit comments

Comments
 (0)