add overlay option

nwlandry · Apr 26, 2024 · 8bdf20c · 8bdf20c
1 parent 971f007
commit 8bdf20c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/plot_fig2.py b/plot_fig2.py
@@ -13,6 +13,8 @@
 measure = "auroc"
 axis_limits = [0, 1]
 
+r0_lines = True
+
 fs.set_fonts()
 fs.set_colors()
 cmap = fs.cmap
@@ -108,8 +110,10 @@ def visualize_networks(i, ax):
     ax.set_yticks([0, 0.5, 1], [0, 0.5, 1])
 
     beta_c = gamma / l
-    for r0 in range(1, 100, 10):
-        ax.plot(var, r0 * beta_c, "-", linewidth=0.25, color="grey")
+
+    if r0_lines:
+        for r0 in range(1, 100, 10):
+            ax.plot(var, r0 * beta_c, "-", linewidth=0.25, color="grey")
 
     if i == 0:
         ax.set_ylabel(r"$\beta$")

diff --git a/plot_fig4.py b/plot_fig4.py
@@ -11,6 +11,8 @@
 gamma = 0.1
 measure = "auroc"
 
+r0_lines = True
+
 axis_limits = [0, 1]
 
 fs.set_fonts()
@@ -113,8 +115,9 @@ def visualize_networks(i, ax):
 
     beta_c = gamma / l
 
-    for r0 in range(1, 100, 10):
-        ax.plot(var, r0 * beta_c, "-", linewidth=0.25, color="grey")
+    if r0_lines:
+        for r0 in range(1, 100, 10):
+            ax.plot(var, r0 * beta_c, "-", linewidth=0.25, color="grey")
 
     if i == 0:
         ax.set_ylabel(r"$\beta$")