This commit is contained in:
Gabriel Rebelski 2021-06-22 16:44:44 +02:00
parent 77c8c106a7
commit 9b63931c30
20 changed files with 2 additions and 2 deletions

View File

@ -10,7 +10,7 @@ class DecisionTree:
def __init__(self, doCreation):
self.data = pd.read_csv("C:\\Users\\Konrad\\PycharmProjects\\Projekt_AI-Automatyczny_saper\\out.csv")
self.data = pd.read_csv("C:\\Users\\Gabriel\\Projekt_AI-Automatyczny_saper\\out.csv")
if doCreation:
self.mapData()
features = ['bomb_type', 'detonation_duration', 'size', 'detonation_area', 'defusable']
@ -21,7 +21,7 @@ class DecisionTree:
decision_tree_model_pkl = open('tree.pkl', 'wb')
pickle.dump(dtree, decision_tree_model_pkl)
decision_tree_model_pkl.close()
decision_tree_model_pkl = open('C:\\Users\\Konrad\\PycharmProjects\\Projekt_AI-Automatyczny_saper\\Engine\\tree.pkl', 'rb')
decision_tree_model_pkl = open('C:\\Users\\Gabriel\\Projekt_AI-Automatyczny_saper\\Engine\\tree.pkl', 'rb')
self.dtree = pickle.load(decision_tree_model_pkl)
def getTree(self):

Binary file not shown.

BIN
tree.pkl Normal file

Binary file not shown.