diff --git a/mcc.py b/mcc.py index 9cee6ed0699b8250ae498f0ce76d429127c32eba..09e3a49a8459c1c7f6ec8142e4674bd200771f77 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