1 files changed, 8 insertions, 6 deletions
diff --git a/src/make_plots.py b/src/make_plots.py
index d28d07e..cbe027e 100644
--- a/src/make_plots.py
+++ b/src/make_plots.py
@@ -49,11 +49,11 @@ 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]
+    x = [50, 100, 500, 1000, 2000, 5000]
+    greedy = [.09, .15, .4, .63, .82, .92]
+    lasso = [.07, .30, .46, .65, 0, 0]
+    max_likel = [.21, .29, .67, .8, .87, .9]
+    sparse_recov = [.25, .32, .7, .82, .89, .92]
     plt.axis((0, 5000, 0, 1))
     plt.xlabel("Number of Cascades")
     plt.ylabel("F1 score")
@@ -68,4 +68,6 @@ def plot_graph(figure_name):
 
 if __name__=="__main__":
     watts_strogatz(n_cascades=5000, lbda=.002, passed_function=
-        convex_optimization.sparse_recovery)
-\ No newline at end of file
+        #convex_optimization.sparse_recovery)
+        #algorithms.greedy_prediction)
+        convex_optimization.type_lasso)
+\ No newline at end of file