Fix KDB algorithm error

2025-08-16 16:15:57 +00:00 · 2022-11-16 11:08:05 +01:00
parent 21814ba01e
commit cad828a21b
1 changed files with 9 additions and 8 deletions
--- a/bayesclass/bayesclass.py
+++ b/bayesclass/bayesclass.py
@@ -293,6 +293,7 @@ class KDB(BayesBase):
            num = 0
            while not exit_cond:
                max_minfo = np.argmax(cond_w[idx, :])
+                if max_minfo in S_nodes:
                    try:
                        dag.add_edge(
                            self.features_[max_minfo], self.features_[idx]