graphs

1 files changed, 4 insertions, 4 deletions

diff --git a/src/make_plots.py b/src/make_plots.py
index 85850b1..c45479e 100644
--- a/src/make_plots.py
+++ b/src/make_plots.py
@@ -71,7 +71,7 @@ def plot_watts_strogatz_graph():
     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="lower right")
+    plt.legend(loc="lower right", fontsize=18)
     ax.set_xticks(x)
     ax.set_xticklabels(tuple(labels))
     plt.savefig("../paper/figures/"+"watts_strogatz.pdf")
@@ -200,7 +200,7 @@ def plot_ROC_curve(figure_name):
     plt.savefig("../paper/figures/"+"ROC_curve.pdf")
 
 
-def plot_kronecker_l2norm_nonsparse():
+def plot_p_init_watts_strogatz():
     plt.clf()
     fig = plt.figure(1)
     x = [.01, .05, .1, .15, .2]
@@ -216,13 +216,13 @@ def plot_kronecker_l2norm_nonsparse():
 
     plt.axis((0, .21, .05, .8))
     plt.xlabel("Number of Cascades", fontsize=20)
-    plt.ylabel("l2-norm", fontsize=20)
+    plt.ylabel("F1-score", 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", fontsize=18)
+    plt.legend(loc="upper right", fontsize=16)
     ax.set_xticks(x)
     ax.set_yticklabels(tuple([.1, .2, .3, .4, .5, .6, .7, .8]), fontsize=20)
     ax.set_xticklabels(tuple(x), fontsize=20)