code/plots/effects_comparison.py

@@ -45,16 +45,12 @@ invert = ["beta_sib_och", "beta_sib_adu"]
 prior_distribution = {effect: np.random.randn(10000) for effect in effects}
 samples = {
     "truth": np.load("output/aggregates_lena_age24_human.npz"),
-    "lena_raw": np.load("output/aggregates_lena_age24_algo.npz"),
-    "vtc_raw": np.load("output/aggregates_vtc_age24_algo.npz"),
+    # "lena_raw": np.load("output/aggregates_lena_age24_algo.npz"),
+    # "vtc_raw": np.load("output/aggregates_vtc_age24_algo.npz"),
     "lena_raw_algo_only": np.load("output/aggregates_lena_age24_algo_only.npz"),
     "vtc_raw_algo_only": np.load("output/aggregates_vtc_age24_algo_only.npz"),
-    "lena_calibrated": np.load(
-        "output/aggregates_lena_age24_dev_siblings_binomial_hurdle_fast.npz"
-    ),
-    "vtc_calibrated": np.load(
-        "output/aggregates_vtc_age24_dev_siblings_binomial_hurdle_fast.npz"
-    ),
+    "lena_calibrated": np.load("output/aggregates_lena_dev.npz"),
+    "vtc_calibrated": np.load("output/aggregates_vtc_dev.npz"),
     "prior": prior_distribution,
 }
 
@@ -62,10 +58,10 @@ samples = {
 labels = {
     "prior": "Prior",
     "truth": "Manual annotations",
-    "lena_raw": "LENA (uncalibrated)",
-    "vtc_raw": "VTC (uncalibrated)",
-    "lena_raw_algo_only": "LENA (uncalibrated, algo only)",
-    "vtc_raw_algo_only": "VTC (uncalibrated, algo only)",
+    # "lena_raw": "LENA (uncalibrated)",
+    # "vtc_raw": "VTC (uncalibrated)",
+    "lena_raw_algo_only": "LENA (uncalibrated)",
+    "vtc_raw_algo_only": "VTC (uncalibrated)",
     "lena_calibrated": "LENA (calibrated)",
     "vtc_calibrated": "VTC (calibrated)",
 }
@@ -73,12 +69,12 @@ labels = {
 positions = {
     "prior": -2,
     "truth": 0,
-    "lena_raw": 3,
-    "vtc_raw": 2,
-    "lena_raw_algo_only": 5,
-    "vtc_raw_algo_only": 4,
-    "vtc_calibrated": 7,
-    "lena_calibrated": 8,
+    # "lena_raw": 3,
+    # "vtc_raw": 2,
+    "lena_raw_algo_only": 3,
+    "vtc_raw_algo_only": 2,
+    "vtc_calibrated": 5,
+    "lena_calibrated": 6,
 }