diff --git a/mcc.py b/mcc.py index 09e3a49a8459c1c7f6ec8142e4674bd200771f77..9cee6ed0699b8250ae498f0ce76d429127c32eba 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,24 +103,29 @@ 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.eraseAllTargets() + ie.eraseAllJointTargets() ie.eraseAllEvidence() - - - - ###### Imputation