fix bug in apple spawning

mvanaltvorst · Jul 8, 2024 · c6931aa · c6931aa
1 parent d60b4b4
commit c6931aa
Show file tree

Hide file tree

Showing 9 changed files with 9,258 additions and 10,603 deletions.
diff --git a/snake-a2c/agent_10x10.state b/snake-a2c/agent_10x10.state
diff --git a/snake-a2c/rewards_10x10.state b/snake-a2c/rewards_10x10.state
diff --git a/snake-a2c/rewards_episodes.svg b/snake-a2c/rewards_episodes.svg
diff --git a/snake-a2c/snake.py b/snake-a2c/snake.py
@@ -69,7 +69,7 @@ def step(self, action):
                     np.random.randint(self.width),
                     np.random.randint(self.height),
                 )
-                if (self.apple_x, self.apple_y) not in self.tail.queue:
+                if (self.apple_x, self.apple_y) not in self.tail.queue and cur_x != self.apple_x and cur_y != self.apple_y:
                     break
         else:
             # Get rid of far tail section

diff --git a/snake-a2c/snake_game_visualisation.gif b/snake-a2c/snake_game_visualisation.gif
diff --git a/snake-a2c/train.ipynb b/snake-a2c/train.ipynb
@@ -186,7 +186,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 5,
    "metadata": {},
    "outputs": [
     {

diff --git a/snake-a2c/visualise.ipynb b/snake-a2c/visualise.ipynb
diff --git a/snake-ppo/snake.py b/snake-ppo/snake.py
@@ -59,7 +59,7 @@ def step(self, action):
                     np.random.randint(self.width),
                     np.random.randint(self.height),
                 )
-                if (self.apple_x, self.apple_y) not in self.tail.queue:
+                if (self.apple_x, self.apple_y) not in self.tail.queue and cur_x != self.apple_x and cur_y != self.apple_y:
                     break
         else:
             # Get rid of far tail section

diff --git a/snake-ppo/visualise.ipynb b/snake-ppo/visualise.ipynb