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