adding figure

1 files changed, 23 insertions, 1 deletions

diff --git a/src/make_plots.py b/src/make_plots.py
index cb78034..d28d07e 100644
--- a/src/make_plots.py
+++ b/src/make_plots.py
@@ -44,6 +44,28 @@ def watts_strogatz(n_cascades, lbda, passed_function):
     algorithms.correctness_measure(G, G_hat, print_values=True)
 
 
+def plot_graph(figure_name):
+    """
+    plot information in a pretty way
+    """
+    plt.clf()
+    x = [100, 500, 1000, 2000, 5000]
+    greedy = [.15, .4, .63, .82, .92]
+    lasso = [.30, .46, .65, 0, 0]
+    max_likel = [.29, .67, .8, .87, .9]
+    sparse_recov = [.32, .7, .82, .89, 0]
+    plt.axis((0, 5000, 0, 1))
+    plt.xlabel("Number of Cascades")
+    plt.ylabel("F1 score")
+    plt.grid(color="lightgrey")
+    plt.plot(x, greedy, 'ko-', color="lightseagreen", label='Greedy')
+    plt.plot(x, lasso, 'ko-', color="orange", label="Lasso")
+    plt.plot(x, max_likel, 'ko-', color="coral", label="MLE")
+    plt.plot(x, sparse_recov, 'ko-', color="k", label="Sparse MLE")
+    plt.legend(loc="lower right")
+    plt.savefig("../paper/figures/"+figure_name)
+
+
 if __name__=="__main__":
-    watts_strogatz(n_cascades=2000, lbda=.002, passed_function=
+    watts_strogatz(n_cascades=5000, lbda=.002, passed_function=
         convex_optimization.sparse_recovery)
\ No newline at end of file