bajo
This commit is contained in:
parent
77c8c106a7
commit
9b63931c30
@ -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.
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