diff --git a/src/AI/DecisionTrees/DecisionTreeBranch.py b/src/AI/DecisionTrees/DecisionTreeBranch.py new file mode 100644 index 0000000..288ca7b --- /dev/null +++ b/src/AI/DecisionTrees/DecisionTreeBranch.py @@ -0,0 +1,10 @@ +from src.AI.DecisionTrees.DecisionTree import DecisionTree + + +class DecisionTreeBranch(object): + subtree: DecisionTree + + def __init__(self, parent, label, subtree): + self.subtree = subtree + self.label = label + self.parent = parent