more changes

1 files changed, 2 insertions, 16 deletions

diff --git a/src/make_plots.py b/src/make_plots.py
index 201e375..947bf57 100644
--- a/src/make_plots.py
+++ b/src/make_plots.py
@@ -44,20 +44,6 @@ def watts_strogatz(n_cascades, lbda, passed_function):
     algorithms.correctness_measure(G, G_hat, print_values=True)
 
 
-# def test():
-#     """
-#     unit test
-#     """
-#     G = cascade_creation.InfluenceGraph(max_proba=1, min_proba=.2)
-#     G.erdos_init(n=50, p=.2)
-#     A = cascade_creation.generate_cascades(G, p_init=.1, n_cascades=1000)
-#     G_hat = algorithms.recovery_passed_function(G, A,
-#                 passed_function=convex_optimization.sparse_recovery,
-#                 floor_cstt=.1, lbda=.001, n_cascades=1000)
-#     algorithms.correctness_measure(G, G_hat, print_values=True)
-
-
 if __name__=="__main__":
-    watts_strogatz(n_cascades=3000, lbda=.002, passed_function=
-        convex_optimization.sparse_recovery)
-        #algorithms.greedy_prediction)
\ No newline at end of file
+    watts_strogatz(n_cascades=500, lbda=.002, passed_function=
+        convex_optimization.sparse_recovery)
\ No newline at end of file