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