tree_increase and alt_dataset merge

This commit is contained in:
Vadzim Valchkovich 2023-05-26 03:26:14 +02:00
parent d97222e9f5
commit dd25c7a145

View File

@ -18,13 +18,13 @@ class Waiter(Object):
self.orientation = copy.copy(state.orientation)
self.basket = copy.copy(state.basket)
self.battery -= state.cost
# self.calcTree()
return state
def dish_in_basket(self, table) -> bool:
return table in self.basket
self.calcTree()
return state
'''
def calcTree(self):
from sklearn import tree
import pandas as pd # for manipulating the csv data
@ -102,9 +102,7 @@ class Waiter(Object):
result = "NO"
print("Predicted Label:", result)
def dampState(self):
self.prev_position = copy.deepcopy(self.position)
self.prev_orientation = copy.copy(self.orientation)
'''
def basket_is_full(self) -> bool:
return self.basket_size == 0