diff options
Diffstat (limited to 'src/algorithms.py')
| -rw-r--r-- | src/algorithms.py | 6 |
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) |