changed createStanfordGraph -> import_from_file & created Cascade class

1 files changed, 1 insertions, 4 deletions

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():
     """