Browse Source

[DATALAD] Recorded changes

Lucas Gautheron 2 months ago
parent
commit
43bd5820cf
1 changed files with 5 additions and 4 deletions
  1. 5 4
      code/models/dev_siblings_combined_binomial_hurdle.stan

+ 5 - 4
code/models/dev_siblings_combined_binomial_hurdle.stan

@@ -92,8 +92,8 @@ parameters {
     matrix<lower=0> [n_recs, n_classes] truth_vocs;
 
     // nuisance parameters
-    array [n_recs] matrix<lower=0>[n_classes,n_classes] actual_confusion_baseline_algo1;
-    array [n_recs] matrix<lower=0>[n_classes,n_classes] actual_confusion_baseline_algo2;
+    array [n_recs] matrix<lower=0,upper=1>[n_classes,n_classes] actual_confusion_baseline_algo1;
+    array [n_recs] matrix<lower=0,upper=1>[n_classes,n_classes] actual_confusion_baseline_algo2;
 
     // confusion parameters
     matrix<lower=1>[n_classes,n_classes] etas_algo1;
@@ -150,10 +150,11 @@ model {
 
     // priors on the nuisance parameters of the confusion model
     for (i in 1:n_classes) {
-        p_algo1[i] ~ beta(3,1);
-        p_algo2[i] ~ beta(3,1);
+        p_algo1[i] ~ beta(3,1); 
         mus_algo1[i,:] ~ uniform(0, 1);
         etas_algo1[i,:] ~ pareto(1, 1.5);
+
+        p_algo2[i] ~ beta(3,1);
         mus_algo2[i,:] ~ uniform(0, 1);
         etas_algo2[i,:] ~ pareto(1, 1.5);