Update feature states setting for datasets

2025-08-15 15:45:54 +00:00 · 2023-02-13 17:34:15 +01:00
parent a5244f1c7f
commit 2ffc06b232
1 changed files with 1 additions and 1 deletions
--- a/bayesclass/clfs.py
+++ b/bayesclass/clfs.py
@@ -513,7 +513,7 @@ class Proposal:
            else self.estimator.default_feature_names(self.Xd.shape[1])
        )
        states = {
-            features[i]: np.unique(self.Xd[:, i]).tolist()
+            features[i]: self.discretizer.get_states_feature(i)
            for i in range(self.Xd.shape[1])
        }
        class_name = (