From 57c967d2bb66ad58f0a7c54127aef096bd5fd890 Mon Sep 17 00:00:00 2001 From: Fize Jacques <jacques.fize@cirad.fr> Date: Tue, 13 Apr 2021 10:46:31 +0200 Subject: [PATCH] debug$ --- eval_mixed_model.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/eval_mixed_model.py b/eval_mixed_model.py index 4df102a..9d10dc3 100644 --- a/eval_mixed_model.py +++ b/eval_mixed_model.py @@ -89,6 +89,9 @@ H = G.copy() float_epsilon = np.finfo(float).eps df_data["p_0"] = df_data.apply(lambda x:1 if G.has_edge(x.u,x.v) else 0,axis =1) for i in range(1,NB_ITERATION+1): + if H.size() < 30: + df_data["p_{0}".format(i)] = df_data["p_{0}".format(i-1)] + continue old_probs = dict(df_data["hash_ p_{0}".format(i-1).split()].values) auc_sbm,auc_spatial = get_aucs(H) if VERBOSE : print("SBM: ",auc_sbm,"SPATIAL: ",auc_spatial) -- GitLab