diff --git a/eval_mixed_model.py b/eval_mixed_model.py index 4df102adec9688693159603ec9349236ccab1871..9d10dc384bd7d17b5d0225f83d2e93d9f38fd511 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)