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