s444409/DSZI_Survival
s444409/DSZI_Survival
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add static method printing decision tree

Browse Source
This commit is contained in:
Michał Czekański 2020-05-21 17:01:27 +02:00
parent 44419b5f28
commit 43b654a6a5
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/AI/DecisionTrees/DecisionTree.py
View File

@ -27,3 +27,14 @@ class DecisionTree(object):
for branch in self.branches: for branch in self.branches:
if branch.label == attr.value: if branch.label == attr.value:
return branch.subtree.giveAnswer(example) return branch.subtree.giveAnswer(example)
@staticmethod
def printTree(tree, depth: int, indent: int = 20):
if isinstance(tree.root, AttributeDefinition):
print("NODE: {}".format(tree.root.name).rjust(indent * depth))
else:
print("NODE: {}".format(str(tree.root)).rjust(indent * depth))
for branch in tree.branches:
print("| {}".format(str(branch.label)).rjust(indent * depth))
DecisionTree.printTree(branch.subtree, depth + 1, indent)