Browse Source

attempt with OCH child effect

Lucas Gautheron 8 months ago
parent
commit
401c0a68ed
1 changed files with 4 additions and 5 deletions
  1. 4 5
      code/models/enumerate_poisson_age.stan

+ 4 - 5
code/models/enumerate_poisson_age.stan

@@ -147,7 +147,7 @@ parameters {
     array [n_recs] vector<lower=0>[n_classes] actual_fp_rate;
 
     matrix<lower=0>[n_children,n_classes] mu_child_level;
-    vector<lower=1>[n_classes] alpha_child_level;
+    vector<lower=0>[n_classes] alpha_child_level;
 
     // confusion parameters
     matrix<lower=0>[n_classes,n_classes] alphas;
@@ -165,7 +165,7 @@ parameters {
     matrix<lower=0>[n_classes,n_corpora] speech_rate_mu;
     vector<lower=0>[n_classes] speech_rate_pop;
     matrix<lower=0> [n_classes,n_rates] speech_rate;
-    matrix<lower=0>[n_children,n_classes] speech_rate_child_level;
+    matrix<lower=0>[n_speech_rate_children,n_classes] speech_rate_child_level;
 
     vector [n_corpora] delta_speech_age;
 }
@@ -195,13 +195,12 @@ model {
     
 
     for (k in 1:n_recs) {
-        // prior on CHI
         (truth_vocs[k,1]/1000/recs_duration) ~ gamma(
             alpha_child_level[1],
             alpha_child_level[1]/(mu_child_level[children[k],1]*exp(delta_speech_age[corpus[children[k]]]*age[k]/12.0))
         );
         
-        (truth_vocs[k,2:]/1000.0/recs_duration) ~ gamma(
+        (truth_vocs[k,2:]/1000/recs_duration) ~ gamma(
             alpha_child_level[2:], alpha_child_level[2:]./mu_child_level[children[k],2:]' //'
         );    
     }
@@ -229,7 +228,7 @@ model {
         lambda_fp[:,i] ~ gamma(alphas_fp[i], alphas_fp[i]/mus_fp[i]);
 
         for (j in 1:n_classes) {
-            mus[i,j] ~ exponential(i==j?2:4);
+            mus[i,j] ~ exponential(i==j?2:8);
             alphas[i,j] ~ gamma(2,1);
             for (c in 1:n_groups) {
                 if (i==1) {