From 635dffadf1a646f4a8e99a97a8c607c5301a9799 Mon Sep 17 00:00:00 2001 From: Fize Jacques <jacques.fize@cirad.fr> Date: Fri, 23 Apr 2021 16:34:42 +0200 Subject: [PATCH] debug --- lib/erosion_model.py | 24 ++++++++++++++++++++++-- 1 file changed, 22 insertions(+), 2 deletions(-) diff --git a/lib/erosion_model.py b/lib/erosion_model.py index 6aa7e7d..19b71c5 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 = False +VERBOSE = True def log(x): if VERBOSE: print(x) @@ -146,10 +146,30 @@ class ErosionModel(): return X_train,X_test,y_train,y_test +def position_str_process(G): + def foo(x): + return [eval(f) for f in re.findall("[-]?\d+.[-]?[\de+-]+", x)] + + is_pos=True + H = G.copy() + for n in list(H.nodes()): + if not "pos" in H.nodes[n]: + is_pos=False + break + if is_pos: + import re + for node in list(G.nodes()): + try: + H.nodes[node]["pos"] = foo(H.nodes[node]["pos"]) + except TypeError: + pass + return H + def eval_erosion_model(G,nb_iter=1,verbose=False): + G = position_str_process(G) erod_mod = ErosionModel(G) erod_mod.erode_n_times(nb_iter) - X_train, X_test, y_train, y_test = erod_mod.get_features() + X_train, X_test, y_train, y_test = erod_mod.get_features(True,True,True) auc_sbm, auc_spa = get_auc_heuristics(G, 60) if verbose:print("SBM: ", auc_sbm, "SPATIAL: ", auc_spa) -- GitLab