diff options
Diffstat (limited to 'src/algorithms.py')
| -rw-r--r-- | src/algorithms.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/algorithms.py b/src/algorithms.py index 24a63f3..43c6721 100644 --- a/src/algorithms.py +++ b/src/algorithms.py @@ -38,7 +38,7 @@ def recovery_passed_function(G, cascades, floor_cstt, passed_function, passed_function should have similar structure to ones in convex_optimation """ G_hat = cascade_creation.InfluenceGraph(max_proba=None) - G_hat.add_nodes_from(G.nodes()) + G_hat.add_nodes_from(G.nodes()) f_x, f_xz = passed_function(*args, **kwargs) @@ -83,8 +83,8 @@ def correctness_measure(G, G_hat, print_values=True): print("Precision: {}".format(precision)) print("Recall: {}".format(recall)) print("F1 score: {}".format(f1_score)) - print("Fall Out: {}".format(fall_out)) - print("l2_norm: {}".format(norm)) + print("Fall Out: {}".format(fall_out)) + print("l2_norm: {}".format(norm)) return fp, fn, tp, tn |