Add method adding branch to decision tree

2020-05-21 16:58:28 +02:00 · 2020-05-21 16:58:28 +02:00 · 9d1299b162
commit 9d1299b162
parent fd54303dba
1 changed files with 4 additions and 0 deletions
--- a/src/AI/DecisionTrees/DecisionTree.py
+++ b/src/AI/DecisionTrees/DecisionTree.py
@ -12,3 +12,7 @@ class DecisionTree(object):
        self.root = root
        self.branches = []
        self.branchesNum = 0
+
+    def addBranch(self, newBranch):
+        self.branches.append(newBranch)
+        self.branchesNum += 1