From 12e52fe9e9e1ac286c1b64342a54d7eccd6d9332 Mon Sep 17 00:00:00 2001 From: jeanpouget-abadie Date: Tue, 3 Feb 2015 10:04:12 -0500 Subject: fixed weird bug? --- src/algorithms.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'src/algorithms.py') 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) -- cgit v1.2.3-70-g09d2