Compare commits
2 Commits
Author | SHA1 | Date | |
---|---|---|---|
7486bd0aa8 | |||
9b63931c30 |
@ -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.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Loading…
Reference in New Issue
Block a user