App.py

@@ -147,7 +147,7 @@ def init_demo(): #Demo purpose
                 pass
                 model = neuralnetwork.loadModel('model_500_hidden.pth')
                 Tractor.drzewo.treeLearn()
-                traktor.snake_move_predict_plant(pole, model, headers=['Coords','Real plant','Predicted plant','Result','Decision'], actions=[traktor.irigate_slot_NN])
+                traktor.snake_move_predict_plant(pole, model, headers=['Coords','Real plant','Predicted plant','Result','Water decision'], actions=[traktor.irigate_slot_NN])
             start_flag=False
         # demo_move()
         old_info=get_info(old_info)

BFS.py

@@ -133,8 +133,12 @@ def check3(tab, state):
     return True
 
 
-def BFS3(istate):
+def BFS3(istate,GT):
+    randomGT=True
+    if(randomGT==True):
         goalTreassuere = (random.randint(0,NUM_X-1), random.randint(0,NUM_Y-1))
+    else:
+        goalTreassuere=GT
     print(goalTreassuere)
     fringe = []
     explored = []