diff options
Diffstat (limited to 'src/make_plots.py')
| -rw-r--r-- | src/make_plots.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/src/make_plots.py b/src/make_plots.py index 3ac8362..c1cafe7 100644 --- a/src/make_plots.py +++ b/src/make_plots.py @@ -23,7 +23,7 @@ def compare_greedy_and_lagrange_cs284r(): #Lagrange Objective G_hat = algorithms.recovery_l1obj_l2constraint(G, A, passed_function=convex_optimization.type_lasso, - floor_cstt=.1, lbda=10) + floor_cstt=.05, lbda=10) algorithms.correctness_measure(G, G_hat, print_values=True) @@ -32,11 +32,11 @@ def test(): unit test """ G = cascade_creation.InfluenceGraph(max_proba=.8) - G.erdos_init(n=20, p=.2) + G.erdos_init(n=30, p=.2) A = cascade_creation.generate_cascades(G, p_init=.1, n_cascades=1000) - G_hat = algorithms.recovery_l1obj_l2constraint(G, A, - passed_function=convex_optimization.sparse_recovery, - floor_cstt=.1, lbda=1) + G_hat = algorithms.recovery_passed_function(G, A, + passed_function=convex_optimization.type_lasso, + floor_cstt=.001, lbda=.0001) algorithms.correctness_measure(G, G_hat, print_values=True) if __name__=="__main__": |