diff --git a/src/AI/DecisionTrees/DecisionTree.py b/src/AI/DecisionTrees/DecisionTree.py index 98d2013..f224bda 100644 --- 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