From 3a2dcd4dd621d8fef681ed84480ed9e9281c7357 Mon Sep 17 00:00:00 2001 From: Maxime Buron <maxime.buron@uca.fr> Date: Thu, 9 Jan 2025 15:33:09 +0100 Subject: [PATCH] fix with joint targets --- mcc.py | 19 +++++++------------ 1 file changed, 7 insertions(+), 12 deletions(-) diff --git a/mcc.py b/mcc.py index 9cee6ed..09e3a49 100755 --- a/mcc.py +++ b/mcc.py @@ -1,4 +1,4 @@ -#!venv/bin/python3 +#!.venv/bin/python3 import pyAgrum as gum import psycopg2 import sys @@ -103,29 +103,24 @@ for row in cursor.fetchall(): if row[i] is None: null_vars.append(v) null_pos.append(i) + ie.addTarget(v) else: ie.addEvidence(v, row[i]) i+=1 - - ie.addJointTarget(set(null_vars)) if len(null_vars) > 0: - # if len(vars) > len(null_vars): - # print(set(null_vars)) - # print(row) - # print(ie.jointTargets()) - # print(ie.targets()) - # else: - # print(vars) - potentiel = ie.jointPosterior(set(null_vars)) if len(vars) > len(null_vars) else ie.evidenceJointImpact(vars,{}) for i in potentiel.loopIn(): inserted_row = instantiation_to_list(i, row, vars) inserted_row.append(row[-1]) inserted_row.append(potentiel.get(i)) cursor.execute(insert_sql, inserted_row) - ie.eraseAllJointTargets() + ie.eraseAllTargets() ie.eraseAllEvidence() + + + + ###### Imputation -- GitLab