fixed weird bug?

1 files changed, 4 insertions, 3 deletions

diff --git a/src/make_plots.py b/src/make_plots.py
index c1cafe7..4a952d1 100644
--- a/src/make_plots.py
+++ b/src/make_plots.py
@@ -31,13 +31,14 @@ def test():
     """
     unit test
     """
-    G = cascade_creation.InfluenceGraph(max_proba=.8)
-    G.erdos_init(n=30, p=.2)
+    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.type_lasso,
-                floor_cstt=.001, lbda=.0001)
+                floor_cstt=.1, lbda=.001)
     algorithms.correctness_measure(G, G_hat, print_values=True)
+    print(G.mat)
 
 if __name__=="__main__":
     test()