Updated preprocessing in JPT

tomsch420 · Dec 31, 2024 · db64750 · db64750
1 parent 1c550d8
commit db64750
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/probabilistic_model/learning/jpt/jpt.py b/src/probabilistic_model/learning/jpt/jpt.py
@@ -181,7 +181,7 @@ def preprocess_data(self, data: pd.DataFrame) -> np.ndarray:
             if isinstance(variable, ScaledContinuous):
                 column = variable.encode(column)
             if isinstance(variable, Symbolic):
-                column = column.apply(lambda x: variable.domain.simple_sets[0].all_elements[x].value)
+                column = column.apply(lambda x: int(x))
             result[:, variable_index] = column
 
         return result

diff --git a/test/test_jax/test_coupling_circuit.py b/test/test_jax/test_coupling_circuit.py
@@ -43,6 +43,7 @@ def generate_parameters(self, x):
     def output_length(self):
         return self.linear.out_features
 
+
 class CouplingCircuitTestCase(unittest.TestCase):
 
     data = jnp.array(np.vstack((np.random.uniform(0, 1, (100, 1)),