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

Fix KDB algorithm error

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2022-11-16 11:08:05 +01:00
parent 21814ba01e
commit cad828a21b
1 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
bayesclass/bayesclass.py
View File

@@ -293,6 +293,7 @@ class KDB(BayesBase):
num = 0 num = 0
while not exit_cond: while not exit_cond:
max_minfo = np.argmax(cond_w[idx, :]) max_minfo = np.argmax(cond_w[idx, :])
if max_minfo in S_nodes:
try: try:
dag.add_edge( dag.add_edge(
self.features_[max_minfo], self.features_[idx] self.features_[max_minfo], self.features_[idx]