This commit is contained in:
Agata Lenz 2020-05-26 19:51:23 +00:00
parent 7e46213a57
commit df1767e243

View File

@ -110,11 +110,12 @@ Funkcja, z której będzie korzystał traktor, aby podjąć decyzję o posadzeni
def decide_to_plant(soil): def decide_to_plant(soil):
if soil.have_plant(): if soil.have_plant():
if soil.get_plant().get_collect(): plant = soil.get_plant()
if plant.collect() == 'True':
info = get_info(soil) info = get_info(soil)
plant.leave_soil() plant.leave_soil()
else: else:
predicted = [['none']] return [['none']]
else: else:
info = get_info(soil) info = get_info(soil)
@ -126,4 +127,6 @@ def decide_to_plant(soil):
return predicted return predicted
``` ```
Funkcja *get_info(soil)* zwraca listę parametrów obiektu Soil, potrzebną do poszukiwań w drzewie decyzyjnym. Pierwszym krokiem jest sprawdzenie, czy w danej ziemi już coś rośnie - jeżeli jest to roślina dojrzała, zostaje ona zebrana, a jej nazwa staje się wartością atrybutu *previous*. Jeżeli nie, nie przewiduje się sadzenia w tym miejscu żadnej rośliny w danym momencie.
Funkcja *get_info(soil)* zwraca listę parametrów obiektu *Soil*, potrzebną do poszukiwań w drzewie decyzyjnym.
W pliku *dataset.py* znajduje się funkcja *create_data_soil()* pozwalająca przetestować działanie algorytmu na obiektach typu *Soil*