fixed weird bug?

1 files changed, 5 insertions, 1 deletions

diff --git a/src/algorithms.py b/src/algorithms.py
index 9ebccf1..0afee24 100644
--- a/src/algorithms.py
+++ b/src/algorithms.py
@@ -11,7 +11,7 @@ def greedy_prediction(G, cascades):
     Returns estimated graph from Greedy algorithm in "Learning Epidemic ..."
     Only works for independent cascade model!
     """
-    G_hat = cascade_creation.InfluenceGraph(max_proba=None)
+    G_hat = cascade_creation.InfluenceGraph(max_proba=None, min_proba=None)
     G_hat.add_nodes_from(G.nodes())
     for node in G_hat.nodes():
         print(node)
@@ -60,6 +60,10 @@ def correctness_measure(G, G_hat, print_values=False):
     """
     edges = set(G.edges())
     edges_hat = set(G_hat.edges())
+
+    print(sorted(edges))
+    print(sorted(edges_hat))
+
     fp = len(edges_hat - edges)
     fn = len(edges - edges_hat)
     tp = len(edges & edges_hat)