From a132c1fe40e31e72db9b1da8173a9920351d6e81 Mon Sep 17 00:00:00 2001 From: Fize Jacques <jacques.fize@cirad.fr> Date: Fri, 23 Apr 2021 14:05:02 +0200 Subject: [PATCH] debug --- lib/erosion_model.py | 1 + lib/link_prediction_eval.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/lib/erosion_model.py b/lib/erosion_model.py index 935737a..6aa7e7d 100644 --- a/lib/erosion_model.py +++ b/lib/erosion_model.py @@ -43,6 +43,7 @@ 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) + if VERBOSE:print(auc_sbm,auc_spatial) edges = get_all_possible_edges(self.H) if auc_sbm > auc_spatial: probs = stochastic_block_model(self.H, edges) diff --git a/lib/link_prediction_eval.py b/lib/link_prediction_eval.py index 5535a9c..c022e83 100644 --- a/lib/link_prediction_eval.py +++ b/lib/link_prediction_eval.py @@ -7,7 +7,7 @@ from evalne.utils import preprocess as pp from .lambda_func import hash_func def get_auc_heuristics(G,timeout=60): - H, _ = pp.prep_graph(G.copy(),maincc=True) + H, _ = pp.prep_graph(G.copy(),maincc=True,relabel=False) traintest_split = LPEvalSplit() traintest_split.compute_splits(H, split_alg="spanning_tree", train_frac=0.90, fe_ratio=1) nee = LPEvaluator(traintest_split) -- GitLab