From a3d47b4320c479c24839f25631abe7d299360644 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Micha=C5=82=20Czeka=C5=84ski?= Date: Thu, 21 May 2020 16:56:20 +0200 Subject: [PATCH] Add DecisionTreeBranch class --- src/AI/DecisionTrees/DecisionTreeBranch.py | 10 ++++++++++ 1 file changed, 10 insertions(+) create mode 100644 src/AI/DecisionTrees/DecisionTreeBranch.py 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