s452686/Super-Saper
s452686/Super-Saper
4
5
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add if __name__ == "__main__" to prevent script from running on import

Browse Source
This commit is contained in:
matixezor 2021-05-18 00:24:58 +02:00
parent d8b165844c
commit ddb75e3d2d
1 changed files with 17 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
src/machine_learning/decision_tree.py
View File

@ -69,16 +69,23 @@ def get_decision(data: dict, root: Node) -> str:
return root.name
def main():
print(RenderTree(tree_root))
print('-' * 150)
score = 0
for test in test_set:
print(f'Test data: {test}')
decision = get_decision(test, tree_root)
print(f'Decision: {decision}')
if decision == test['action']:
score += 1
print(f'Accuracy: {score/len(test_set)}')
tree_root = tree_learn(training_set, attribs, 'detonation')
print(RenderTree(tree_root))
print('-' * 150)
score = 0
for test in test_set:
print(f'Test data: {test}')
decision = get_decision(test, tree_root)
print(f'Decision: {decision}')
if decision == test['action']:
score += 1
print(f'Accuracy: {score/len(test_set)}')
if __name__ == "__main__":
main()