Minor changes

1 files changed, 3 insertions, 4 deletions

diff --git a/qlearning.py b/qlearning.py
index 46d19d9..6e6d54c 100755
--- a/qlearning.py
+++ b/qlearning.py
@@ -131,10 +131,9 @@ class QTable:
 perf=0
 last_state=None
 last_action=None
-game=Snake(length=4,fps=300,startat=(10,10))
+game=Snake(length=1,fps=300,startat=(random.randint(0,29),random.randint(0,29)))
 qtable=QTable("qtable.txt")
-
-for i in range(0,10000):
+while True:
     result=0
     stuck=0
     state=qtable.get_state(game)
@@ -144,7 +143,7 @@ for i in range(0,10000):
         new_state=qtable.get_state(game)
 
         # Agent is stuck
-        if stuck>=game.grid_width*game.grid_height:
+        if stuck>=(game.grid_width*game.grid_height)/2:
             game.new_game()
             break