From d08aea4681a9433dfc5dd9f5bc7c2a433a83f1fe Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana=20G=C3=B3mez?= Date: Thu, 12 Jan 2023 14:05:27 +0100 Subject: [PATCH] fix AODE state_names mistake --- bayesclass/clfs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bayesclass/clfs.py b/bayesclass/clfs.py index e68d435..e0eb4cc 100644 --- a/bayesclass/clfs.py +++ b/bayesclass/clfs.py @@ -348,6 +348,7 @@ class AODE(BayesBase, BaseEnsemble): """Build SPODE estimators (Super Parent One Dependent Estimator)""" self.models_ = [] class_edges = [(self.class_name_, f) for f in self.features_] + states = dict(state_names=kwargs.pop("state_names", [])) for idx in range(len(self.features_)): feature_edges = [ (self.features_[idx], f) @@ -358,7 +359,6 @@ class AODE(BayesBase, BaseEnsemble): model = BayesianNetwork( feature_edges, show_progress=self.show_progress ) - states = dict(state_names=kwargs.pop("state_names", [])) model.fit( self.dataset_, estimator=BayesianEstimator,