diff --git a/lib/erosion_model.py b/lib/erosion_model.py index 152ebaf8b7da2c6a1832175671a99d7d34e76c4b..935737ac8d238632f302912a52d16f286cf576dc 100644 --- a/lib/erosion_model.py +++ b/lib/erosion_model.py @@ -15,7 +15,7 @@ import networkx as nx import numpy as np float_epsilon = np.finfo(float).eps -VERBOSE = True +VERBOSE = False def log(x): if VERBOSE: print(x) @@ -43,7 +43,6 @@ class ErosionModel(): old_probs = dict(self.probs_df["hash_ p_{0}".format(self.nb_of_erosion - 1).split()].values) auc_sbm, auc_spatial = get_auc_heuristics(self.H, 60) - print(auc_spatial,auc_sbm) edges = get_all_possible_edges(self.H) if auc_sbm > auc_spatial: probs = stochastic_block_model(self.H, edges) @@ -65,7 +64,6 @@ class ErosionModel(): [(1 / self.H.size()) - ((probs_dom[ix]*sum_prob_dom)/sum_prob_dom_H) for ix, ed in enumerate(edges) if self.H.has_edge(*ed)]) hhh[hhh < 0] = 0 new_nb_edges = hhh.sum() * self.H.size() - #print(hhh) probs_erosion = np.asarray([old_probs[hash_func(ed)] - probs_dom[ix] for ix, ed in enumerate(edges)]) probs_erosion[probs_erosion <= 0] = float_epsilon