diff options
Diffstat (limited to 'src/make_plots.py')
| -rw-r--r-- | src/make_plots.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/make_plots.py b/src/make_plots.py index 18d0874..57c5caa 100644 --- a/src/make_plots.py +++ b/src/make_plots.py @@ -31,12 +31,12 @@ def test(): """ unit test """ - G = cascade_creation.InfluenceGraph(max_proba=.3) - G.erdos_init(n=10, p=.2) - A = cascade_creation.generate_cascades(G, p_init=.2, n_cascades=100) + G = cascade_creation.InfluenceGraph(max_proba=.8) + G.erdos_init(n=100, p=.1) + A = cascade_creation.generate_cascades(G, p_init=.02, n_cascades=1000) G_hat = algorithms.recovery_l1obj_l2constraint(G, A, passed_function=convex_optimization.sparse_recovery, - floor_cstt=.1, lbda=10) + floor_cstt=.1, lbda=1) algorithms.correctness_measure(G, G_hat, print_values=True) if __name__=="__main__": |