Update figure labels for 3subj sim

emdupre · Jun 22, 2024 · 4da9302 · 4da9302
1 parent 66cb738
commit 4da9302
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/experiments/three_subjects.py b/experiments/three_subjects.py
@@ -172,16 +172,17 @@ def main(seed, mu_pri, sigma_pri, sigma_subj, sigma_obs, num_timesteps, hazard_p
     axs["A"].text(
         0.0,
         0.5,
-        "True",
+        "Initial",
         transform=(
             axs["A"].transAxes + ScaledTranslation(-50 / 72, +2 / 72, fig.dpi_scale_trans)
         ),
         size="large",
         va="center",
         rotation=90,
     )
-    axs["D"].plot(jnp.average(signals, axis=0), ls=(0, (5, 1)), c="#7F7776")
-    axs["D"].set_title("$\mu^0$", size="x-large")
+    axs["D"].plot(jnp.average(obs, axis=0), ls=(0, (5, 1)), c="#7F7776")
+    # axs["D"].plot(jnp.average(signals, axis=0), ls=(0, (5, 1)), c="#7F7776")
+    axs["D"].set_title("Group \n estimate", size="large")
 
     for i, (ax, c) in enumerate(
         zip([axs["E"], axs["F"], axs["G"]], ["#017E7C", "#016895", "#006B81"])
@@ -191,7 +192,7 @@ def main(seed, mu_pri, sigma_pri, sigma_subj, sigma_obs, num_timesteps, hazard_p
     axs["E"].text(
         0.0,
         0.5,
-        "Sampled",
+        "Recovered",
         transform=(
             axs["E"].transAxes + ScaledTranslation(-50 / 72, +2 / 72, fig.dpi_scale_trans)
         ),