From e053757bcf2c13825c9f848b01df297c4045d658 Mon Sep 17 00:00:00 2001 From: Adam Mikolajczak Date: Mon, 10 Jun 2024 14:34:38 +0200 Subject: [PATCH] fix: ziemniaki problem Added more examples to the learning set of decision tree. --- src/tile.py | 3 --- src/tree.csv | 9 +++++++++ 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/src/tile.py b/src/tile.py index 28498ff8..51877c32 100644 --- a/src/tile.py +++ b/src/tile.py @@ -97,9 +97,6 @@ class Tile(pygame.sprite.Sprite): result = classes[predicted.item()] - - if result == "ziemniak": - result = 'marchew' return result diff --git a/src/tree.csv b/src/tree.csv index 37b9f61e..194f660a 100644 --- a/src/tree.csv +++ b/src/tree.csv @@ -222,3 +222,12 @@ grass,10,no_plant,marchew,dynia,grass,water,plant(seler) grass,10,no_plant,water,dynia,cebula,cebula,plant(seler) grass,10,no_plant,ogórek,dynia,cebula,grass,plant(seler) grass,10,no_plant,szpinak,dynia,fasola,grass,plant(seler) +ziemniak,40,planted,water,grass,szpinak,grass,move +ziemniak,50,grown,szpinak,szpinak,grass,grass,move +ziemniak,10,planted,water,ziemniak,seler,water,move +ziemniak,20,planted,szpinak,szpinak,grass,seler,move +ziemniak,30,planted,grass,water,ziemniak,grass,move +ziemniak,40,planted,growing,grass,grass,burak,move +ziemniak,50,grown,growing,szpinak,water,grass,move +ziemniak,10,planted,growing,water,szpinak,marchew,move +ziemniak,20,planted,growing,szpinak,grass,kapusta,move