From 2ffc06b232d935013d0c6ced82ac4bf1af5032f3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana?= Date: Mon, 13 Feb 2023 17:34:15 +0100 Subject: [PATCH] Update feature states setting for datasets --- bayesclass/clfs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bayesclass/clfs.py b/bayesclass/clfs.py index 0126c36..7db719b 100644 --- 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 = (