Fix KDB algorithm error

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