acfx 0.3.7.dev1__tar.gz → 0.3.7.dev2__tar.gz

{acfx-0.3.7.dev1 → acfx-0.3.7.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: acfx
-Version: 0.3.7.dev1
+Version: 0.3.7.dev2
 Summary: Actionable Counterfactual eXplanations
 Author-email: Szymon Bobek <szymon.bobek@gmail.com>, Piotr Kubacki <piotr.kubacki00@gmail.com>
 License: MIT

{acfx-0.3.7.dev1 → acfx-0.3.7.dev2}/acfx/evaluation/loss.py

@@ -27,13 +27,30 @@ def compute_causal_penalty_bayesian_adjacency(samples, model : DiscreteBayesianN
             cpd = model.get_cpds(node)
             if not cpd:
                 continue
+            try:
+                node_idx = nodes.index(node)
+                node_value = int(np.round(sample[node_idx]))
 
-            # Get parent values for this sample
-            parent_values = [int(sample[nodes.index(p)]) for p in parents]
-            node_value = sample[nodes.index(node)]
+                # Get parent values for this sample
+                parent_values = [int(np.round(sample[nodes.index(p)])) for p in parents]
 
-            # Find probability from CPD
-            prob = cpd.get_value(**{node: int(node_value), **dict(zip(parents, parent_values))})
+                node_cardinality = cpd.variable_card
+
+                node_value = max(0, min(node_value, node_cardinality - 1))
+
+                for i, p_name in enumerate(parents):
+                    parent_cardinality = model.get_cardinality(p_name)
+                    parent_values[i] = max(0, min(parent_values[i], parent_cardinality - 1))
+
+                # Find probability from CPD
+                kwargs = {node: node_value}
+                for p_name, p_val in zip(parents, parent_values):
+                    kwargs[p_name] = p_val
+
+                prob = cpd.get_value(**kwargs)
+            except (IndexError, ValueError) as e:
+                prob = 1e-8
+                print(f'compute_causal_penalty_bayesian_adjacency: {e}')
             if np.abs(prob) < 1e-8:
                 prob = 1e-8
             sample_penalty += -np.log(prob)

{acfx-0.3.7.dev1 → acfx-0.3.7.dev2}/acfx.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: acfx
-Version: 0.3.7.dev1
+Version: 0.3.7.dev2
 Summary: Actionable Counterfactual eXplanations
 Author-email: Szymon Bobek <szymon.bobek@gmail.com>, Piotr Kubacki <piotr.kubacki00@gmail.com>
 License: MIT

{acfx-0.3.7.dev1 → acfx-0.3.7.dev2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "acfx"
-version = "0.3.7dev1"
+version = "0.3.7dev2"
 description = "Actionable Counterfactual eXplanations"
 readme = "README.md"
 authors = [