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