improved plotting

HuangHam · Nov 25, 2019 · 2d30177 · 2d30177
1 parent 51140e3
commit 2d30177
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/bin/analyzeDepth.py b/bin/analyzeDepth.py
@@ -103,6 +103,16 @@ def loadCheckpoint(r):
 
             plot.figure(figsize=(4,2.5))
             plot.scatter(X,Y,color=C,alpha=0.5)
+            plot.ylabel("% Test Solved",
+                    fontsize=FONTSIZE)
+            plot.xlabel({"MAX": "Max depth",
+                         "MEAN": "Avg. depth",
+                         "SIZE": "Library size"}[mode],
+                        fontsize=FONTSIZE)
+            plot.xticks(fontsize=TICKFONTSIZE)
+            plot.yticks(fontsize=TICKFONTSIZE)
+            plot.gca().spines['right'].set_visible(False)
+            plot.gca().spines['top'].set_visible(False)        
             plot.savefig(f"figures/depthVersusAccuracy_revision_{mode}.png")
         sys.exit()