Doctorado-ML/bayesclass
1
0
Fork 0
mirror of https://github.com/Doctorado-ML/bayesclass.git synced 2025-08-16 16:15:57 +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

17
bayesclass/bayesclass.py
View File

@@ -293,14 +293,15 @@ 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, :])
try: if max_minfo in S_nodes:
dag.add_edge( try:
self.features_[max_minfo], self.features_[idx] dag.add_edge(
) self.features_[max_minfo], self.features_[idx]
num += 1 )
except ValueError: num += 1
# Loops are not allowed except ValueError:
pass # Loops are not allowed
pass
cond_w[idx, max_minfo] = -1 cond_w[idx, max_minfo] = -1
exit_cond = num == n_edges or np.all(cond_w[idx, :] <= 0) exit_cond = num == n_edges or np.all(cond_w[idx, :] <= 0)