acfx 0.3.7.dev1__py3-none-any.whl → 0.3.7.dev2__py3-none-any.whl

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.dist-info → acfx-0.3.7.dev2.dist-info}/METADATA

@@ -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.dist-info → acfx-0.3.7.dev2.dist-info}/RECORD

@@ -12,11 +12,11 @@ acfx/evaluation/LogisticRegressionCounterOptimizer.py,sha256=LqumbdlzrPi4p0XlSqI
 acfx/evaluation/__init__.py,sha256=dTLbavEOvI77m9-jkxF_GdsCpTgpGcbu6BooGA_Gsrs,460
 acfx/evaluation/bayesian_model.py,sha256=ohm0cnl6HYg8QetloDwDgrGnSkaacvrU7op31EjBiYM,3153
 acfx/evaluation/ccfs.py,sha256=sEcfxaMlxSUnGudnrMEzc3eroKG6jyrbCRaWZ1Vn6Ow,19570
-acfx/evaluation/loss.py,sha256=FLT7oEOzy0_ghYcqGPTWz141WLR53LS5TNgQbexPV5c,8888
+acfx/evaluation/loss.py,sha256=UmC9oEx2iSEH2CRZEooW8_NnvMEvRU7i4clOwUv_740,9585
 acfx/evaluation/multi_dataset_evaluation.py,sha256=BVXjio65Hp0ayVrxs4aMwH0hWs79RkZHV6Qmk2fuS7I,2285
 acfx/evaluation/utils.py,sha256=389r9933X6y-XIfXl0NAtJ7amcWlkbQDw7CLuJOgLj4,3607
-acfx-0.3.7.dev1.dist-info/licenses/LICENSE,sha256=gZfIb-6rTt3sJfwq84Qz-PiXKvOuTtnuQWyRqT9_yk0,1069
-acfx-0.3.7.dev1.dist-info/METADATA,sha256=Hj98s7QJDqAoi2POZYDNuk7XLA6h1yS5IW4caRY1T3M,2587
-acfx-0.3.7.dev1.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-acfx-0.3.7.dev1.dist-info/top_level.txt,sha256=PrYErhyLTFcKay5Vi6hyl2C3iVxtlzG81xMY7ogRjkU,5
-acfx-0.3.7.dev1.dist-info/RECORD,,
+acfx-0.3.7.dev2.dist-info/licenses/LICENSE,sha256=gZfIb-6rTt3sJfwq84Qz-PiXKvOuTtnuQWyRqT9_yk0,1069
+acfx-0.3.7.dev2.dist-info/METADATA,sha256=xZwAG5AEhCltk1yZInwVq1SMefBrXjrToJR9SRyLcLk,2587
+acfx-0.3.7.dev2.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+acfx-0.3.7.dev2.dist-info/top_level.txt,sha256=PrYErhyLTFcKay5Vi6hyl2C3iVxtlzG81xMY7ogRjkU,5
+acfx-0.3.7.dev2.dist-info/RECORD,,