diff options
Diffstat (limited to 'src')
| -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 9b1fdf4..5a61c10 100644 --- a/src/make_plots.py +++ b/src/make_plots.py @@ -39,8 +39,8 @@ def compare_greedy_and_lagrange_cs284r(): for the CS284r project """ G = cascade_creation.InfluenceGraph(max_proba = .8) - G.import_from_file("subset_facebook_SNAP.txt") - A = cascade_creation.generate_cascades(G, p_init=.05, n_cascades=100) + G.import_from_file("../datasets/subset_facebook_SNAPnormalize.txt") + A = cascade_creation.generate_cascades(G, p_init=.05, n_cascades=1000) #Greedy G_hat = algorithms.greedy_prediction(G, A) |