ソースを参照

[DATALAD] Recorded changes

Lucas Gautheron 2 ヶ月 前
コミット
541cf78566
1 ファイル変更8 行追加2 行削除
  1. 8 2
      code/models/dev_siblings_combined_binomial_hurdle.stan

+ 8 - 2
code/models/dev_siblings_combined_binomial_hurdle.stan

@@ -60,6 +60,8 @@ transformed data {
     int no_siblings = 0;
     int has_siblings = 0;
 
+    real p_outlier = 0.025;
+
     for (c in 1:n_clips) {
         recording_age[group[c]] = clip_age[c];
     }
@@ -160,8 +162,12 @@ model {
 
         for (j in 1:n_classes) {
             for (c in 1:n_groups) {
-                lambda_algo1[c,i,j] ~ beta_proportion(mus_algo1[i,j], etas_algo1[i,j]);
-                lambda_algo2[c,i,j] ~ beta_proportion(mus_algo2[i,j], etas_algo2[i,j]);
+                // tolerance to outliers
+                target += log_mix(
+                    p_outlier,
+                    0,
+                    beta_proportion_lpdf(lambda_algo1[c,i,j] | mus_algo1[i,j], etas_algo1[i,j]) + beta_proportion(lambda_algo2[c,i,j] | mus_algo2[i,j], etas_algo2[i,j])
+                );
             }
         }
     }