Skip to content
Snippets Groups Projects
Commit 3a2dcd4d authored by Maxime BURON's avatar Maxime BURON
Browse files

fix with joint targets

parent 3fdd1fc8
No related branches found
No related tags found
No related merge requests found
#!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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment