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):
if soil.have_plant():
if soil.get_plant().get_collect():
plant = soil.get_plant()
if plant.collect() == 'True':
info = get_info(soil)
plant.leave_soil()
else:
predicted = [['none']]
return [['none']]
else:
info = get_info(soil)
@ -126,4 +127,6 @@ def decide_to_plant(soil):
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*