s452711/Projekt_Sztuczna_Inteligencja
5
3
Fork 0
Code Issues Pull Requests Releases Wiki Activity

big commit (ostroznie!)

Browse Source
This commit is contained in:
s452645 2021-05-26 13:47:02 +02:00
parent 32c50f27fc
commit 9a910a9284
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
algorithms/learn/decision_tree.py
View File

@ -1,5 +1,6 @@
import os import os
import json import json
from matplotlib import pyplot
from joblib import dump, load from joblib import dump, load
from sklearn import tree from sklearn import tree
from sklearn.feature_extraction import DictVectorizer from sklearn.feature_extraction import DictVectorizer
@ -38,6 +39,14 @@ class DecisionTree:
# print a tree (not necessary) # print a tree (not necessary)
print(tree.export_text(self.clf, feature_names=self.vec.get_feature_names())) print(tree.export_text(self.clf, feature_names=self.vec.get_feature_names()))
# plot a tree (not necessary)
fig = pyplot.figure(figsize=(50, 40))
_ = tree.plot_tree(self.clf,
feature_names=self.vec.get_feature_names(),
class_names=self.clf.classes_,
filled=True)
fig.savefig("decistion_tree.png")
def save(self): def save(self):
dump(self.clf, 'decision_tree.joblib') dump(self.clf, 'decision_tree.joblib')
dump(self.vec, 'dict_vectorizer.joblib') dump(self.vec, 'dict_vectorizer.joblib')
@ -79,4 +88,3 @@ if __name__ == "__main__":
decision_tree = DecisionTree() decision_tree = DecisionTree()
decision_tree.build("training_set.txt", 15) decision_tree.build("training_set.txt", 15)
decision_tree.test() decision_tree.test()
decision_tree.save()

1
main.py
View File

@ -126,6 +126,7 @@ def main():
game.agent_take_last_action() game.agent_take_last_action()
game.cleanup_after_game_loop() game.cleanup_after_game_loop()
game.draw_minefield() game.draw_minefield()
game.run_in_game_menu_overlay(pygame.mouse.get_pos(), events)
pygame.display.flip() pygame.display.flip()
if auto: if auto: