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

Default indent fix in decision tree print

Browse Source
This commit is contained in:
Michał Czekański 2020-05-24 00:38:12 +02:00
parent 2fbc877141
commit 0d6b3e4e84
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -29,7 +29,7 @@ class DecisionTree(object):
return branch.subtree.giveAnswer(example) return branch.subtree.giveAnswer(example)
@staticmethod @staticmethod
def printTree(tree, depth: int, indent: int = 20): def printTree(tree, depth: int, indent: int = 50):
if isinstance(tree.root, AttributeDefinition): if isinstance(tree.root, AttributeDefinition):
print("NODE: {}".format(tree.root.name).rjust(indent * depth)) print("NODE: {}".format(tree.root.name).rjust(indent * depth))
else: else: