Doctorado-ML/bayesclass
1
0
Fork 0
mirror of https://github.com/Doctorado-ML/bayesclass.git synced 2025-08-15 23:55:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix AODE state_names mistake

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-01-12 14:05:27 +01:00
parent dd2e0a3b7e
commit d08aea4681
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
bayesclass/clfs.py
View File

@@ -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,