From 3fdd1fc851b4a34f9dcb5252a12de6d0603a2a49 Mon Sep 17 00:00:00 2001
From: Maxime Buron <maxime.buron@uca.fr>
Date: Thu, 9 Jan 2025 15:28:08 +0100
Subject: [PATCH] fix with joint targets
---
mcc.py | 19 ++++++++++++-------
1 file changed, 12 insertions(+), 7 deletions(-)
diff --git a/mcc.py b/mcc.py
index 09e3a49..9cee6ed 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
--
GitLab