Fix the bug in the discretize method

erdogant · Nov 15, 2023 · c1e4f27 · c1e4f27
1 parent 851a00d
commit c1e4f27
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/bnlearn/discretize/discretize.py b/bnlearn/discretize/discretize.py
@@ -46,6 +46,8 @@ def discretize(
         verbose=verbose,
     )
 
+    continuous_columns = sorted(continuous_columns, key=lambda x: data.columns.get_loc(x))
+
     # Extend the breaks to the left with 1% to deal with (open,closed] intervals
     for i, col in enumerate(continuous_columns):
         breaks = continuous_edges[i]