summaryrefslogtreecommitdiff
path: root/qlearning.py
diff options
context:
space:
mode:
Diffstat (limited to 'qlearning.py')
-rwxr-xr-xqlearning.py7
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