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