something plot

1 files changed, 9 insertions, 8 deletions

diff --git a/src/make_plots.py b/src/make_plots.py
index c5a50f6..9615d1b 100644
--- a/src/make_plots.py
+++ b/src/make_plots.py
@@ -193,7 +193,7 @@ def plot_ROC_curve(figure_name):
             color="orange", label="Lasso-50 cascades")
     ax.plot(recall_sparse_50, precision_sparse_50, 'ko-', 
             color="cornflowerblue", label="Our Method-50 cascades")
-    plt.legend(loc="upper right")
+    plt.legend(loc="upper right", fontsize=14)
     plt.savefig("../paper/figures/"+"ROC_curve.pdf")
 
 
@@ -208,20 +208,21 @@ def plot_kronecker_l2norm_nonsparse():
 
     fig, ax = plt.subplots()
 
-    #plt.subplots_adjust(bottom=.2, top=.85)
-    #plt.xticks(ha="right", rotation=45)
+    plt.subplots_adjust(bottom=.2, top=.85)
+    plt.xticks(ha="right", rotation=45)
 
-    #plt.axis((50, 2000, 0, 145))
-    plt.xlabel("Number of Cascades")
-    plt.ylabel("l2-norm")
+    plt.axis((0, .21, .05, .8))
+    plt.xlabel("Number of Cascades", fontsize=20)
+    plt.ylabel("l2-norm", fontsize=20)
     plt.grid(color="lightgrey")
     ax.plot(x, greedy, 'ko-', color="lightseagreen", label='Greedy')
     ax.plot(x, lasso, 'ko-', color="orange", label="Lasso")
     ax.plot(x, max_likel, 'ko-', color="cornflowerblue", label="MLE")
     ax.plot(x, sparse_recov, 'ko-', color="k", label="Our Method")
-    plt.legend(loc="upper right")
+    plt.legend(loc="upper right", fontsize=18)
     ax.set_xticks(x)
-    ax.set_xticklabels(tuple(x))
+    ax.set_yticklabels(tuple([.1, .2, .3, .4, .5, .6, .7, .8]), fontsize=20)
+    ax.set_xticklabels(tuple(x), fontsize=20)
     plt.savefig("../paper/figures/"+"watts_strogatz_p_init.pdf")