s474137/automatyczny_kelner
3
3
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

tree_increase and alt_dataset merge

Browse Source
This commit is contained in:
Vadzim Valchkovich 2023-05-26 03:26:14 +02:00
parent d97222e9f5
commit dd25c7a145
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/obj/Waiter.py
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
@ -101,10 +101,8 @@ class Waiter(Object):
else: else:
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