aboutsummaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/make_plots.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/make_plots.py b/src/make_plots.py
index 5a61c10..7c8bebb 100644
--- a/src/make_plots.py
+++ b/src/make_plots.py
@@ -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=50)
#Greedy
G_hat = algorithms.greedy_prediction(G, A)