sticking with AUPRC

nwlandry · Apr 23, 2024 · 429fa5b · 429fa5b
1 parent ba0acb3
commit 429fa5b
Show file tree

Hide file tree

Showing 12 changed files with 8 additions and 8 deletions.
diff --git a/Figures/Fig1/fig1.pdf b/Figures/Fig1/fig1.pdf
diff --git a/Figures/Fig1/fig1.png b/Figures/Fig1/fig1.png
diff --git a/Figures/Fig2/fig2.pdf b/Figures/Fig2/fig2.pdf
diff --git a/Figures/Fig2/fig2.png b/Figures/Fig2/fig2.png
diff --git a/Figures/Fig4/fig4.pdf b/Figures/Fig4/fig4.pdf
diff --git a/Figures/Fig4/fig4.png b/Figures/Fig4/fig4.png
diff --git a/Figures/Fig5/fig5.pdf b/Figures/Fig5/fig5.pdf
diff --git a/Figures/Fig5/fig5.png b/Figures/Fig5/fig5.png
diff --git a/plot_fig1.py b/plot_fig1.py
@@ -212,7 +212,7 @@
     va="top",
 )
 
-measure = "auroc"
+measure = "auprc"
 
 with open("Data/zkc_infer_vs_tmax.json") as file:
     data = json.load(file)
@@ -253,8 +253,8 @@
         r"$\mathregular{10^4}$",
     ],
 )
-ax3.set_ylim([0.5, 1])
-ax3.set_yticks([0.5, 0.75, 1], [0.5, 0.75, 1])
+ax3.set_ylim([0, 1])
+ax3.set_yticks([0, 0.5, 1], [0, 0.5, 1])
 
 ax3.legend(
     loc="lower right",
@@ -293,7 +293,7 @@
     extent=(min(frac), max(frac), max(beta), min(beta)),
     aspect="auto",
     cmap=cmap,
-    vmin=0.5,
+    vmin=0,
     vmax=1,
 )
 ax4.set_xlabel(r"Complexity, $\lambda$")
@@ -305,7 +305,7 @@
 cbar_ax = fig.add_axes([0.91, 0.15, 0.015, 0.31])  # x, y, width, height
 cbar = plt.colorbar(c, cax=cbar_ax)
 cbar.set_label(measure.upper(), fontsize=axislabel_fontsize, rotation=270, labelpad=10)
-cbar_ax.set_yticks([0.5, 1], [0.5, 1], fontsize=tick_fontsize)
+cbar_ax.set_yticks([0, 1], [0, 1], fontsize=tick_fontsize)
 
 plt.savefig("Figures/Fig1/fig1.png", dpi=1000)
 plt.savefig("Figures/Fig1/fig1.pdf", dpi=1000)

diff --git a/plot_fig2.py b/plot_fig2.py
@@ -10,7 +10,7 @@
 
 axis_limits = [0, 1]
 
-measure = "auroc"
+measure = "auprc"
 
 fs.set_fonts()
 fs.set_colors()

diff --git a/plot_fig4.py b/plot_fig4.py
@@ -8,7 +8,7 @@
 import fig_settings as fs
 from lcs import *
 
-measure = "auroc"
+measure = "auprc"
 axis_limits = [0, 1]
 
 axislabel_fontsize = 20

diff --git a/plot_fig5.py b/plot_fig5.py
@@ -8,7 +8,7 @@
 import fig_settings as fs
 from lcs import *
 
-measure = "auroc"
+measure = "auprc"
 
 axis_limits = [0, 1]