From 6c2ff72156808975b2434e27a70b03cd54fdaecf Mon Sep 17 00:00:00 2001 From: jeanpouget-abadie Date: Sat, 29 Nov 2014 14:45:11 -0500 Subject: changed createStanfordGraph -> import_from_file & created Cascade class --- jpa_test/algorithms.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) (limited to 'jpa_test/algorithms.py') diff --git a/jpa_test/algorithms.py b/jpa_test/algorithms.py index 2ee08ec..9973dc3 100644 --- a/jpa_test/algorithms.py +++ b/jpa_test/algorithms.py @@ -8,13 +8,10 @@ def greedy_prediction(G, cascades): returns estimated graph """ G_hat = cascade_creation.InfluenceGraph(max_proba=None) - G.add_nodes_from(G.nodes()) + G_hat.add_nodes_from(G.nodes()) for node in G.nodes(): unaccounted = cascades for cascade in cascades: - time_step = 0 - while not cascade[time_step][node]: - time_step += 1 def test(): """ -- cgit v1.2.3-70-g09d2