diff options
Diffstat (limited to 'src/make_plots.py')
| -rw-r--r-- | src/make_plots.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/make_plots.py b/src/make_plots.py index 5a61c10..dfe9c86 100644 --- a/src/make_plots.py +++ b/src/make_plots.py @@ -16,7 +16,7 @@ def plot_rip_numberofnodes(max_proba, n_min, n_max, p_init, n_cascades, K_max): for n_nodes in x: print n_nodes G = cascade_creation.InfluenceGraph(max_proba=.3) - G.erdos_init(n=n_nodes, p=.1) #TODO: handle different inits! + G.erdos_init(n=n_nodes, p=.1) # TODO: handle different inits! cascades = cascade_creation.generate_cascades(G, p_init=p_init, n_cascades=n_cascades) M, __ = cascade_creation.icc_matrixvector_for_node(cascades, None) @@ -40,7 +40,7 @@ def compare_greedy_and_lagrange_cs284r(): """ G = cascade_creation.InfluenceGraph(max_proba = .8) G.import_from_file("../datasets/subset_facebook_SNAPnormalize.txt") - A = cascade_creation.generate_cascades(G, p_init=.05, n_cascades=1000) + A = cascade_creation.generate_cascades(G, p_init=.05, n_cascades=2000) #Greedy G_hat = algorithms.greedy_prediction(G, A) |