Compare commits

...

2 Commits

Author SHA1 Message Date
7486bd0aa8 usuniety plik drzewa 2021-06-22 16:57:13 +02:00
9b63931c30 bajo 2021-06-22 16:44:44 +02:00
19 changed files with 2 additions and 2 deletions

View File

@ -10,7 +10,7 @@ class DecisionTree:
def __init__(self, doCreation): 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: if doCreation:
self.mapData() self.mapData()
features = ['bomb_type', 'detonation_duration', 'size', 'detonation_area', 'defusable'] features = ['bomb_type', 'detonation_duration', 'size', 'detonation_area', 'defusable']
@ -21,7 +21,7 @@ class DecisionTree:
decision_tree_model_pkl = open('tree.pkl', 'wb') decision_tree_model_pkl = open('tree.pkl', 'wb')
pickle.dump(dtree, decision_tree_model_pkl) pickle.dump(dtree, decision_tree_model_pkl)
decision_tree_model_pkl.close() 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) self.dtree = pickle.load(decision_tree_model_pkl)
def getTree(self): def getTree(self):

Binary file not shown.