summaryrefslogtreecommitdiff
path: root/qlearning.py
diff options
context:
space:
mode:
Diffstat (limited to 'qlearning.py')
-rwxr-xr-xqlearning.py23
1 files changed, 23 insertions, 0 deletions
diff --git a/qlearning.py b/qlearning.py
new file mode 100755
index 0000000..7834f6f
--- /dev/null
+++ b/qlearning.py
@@ -0,0 +1,23 @@
+#!/usr/bin/env python
+import sys
+import numpy as np
+
+# Import snake game
+from snake import Snake
+
+
+
+# Setup QTable
+qtable=np.zeros((16, 4))
+
+
+
+game=Snake()
+
+def event_handler(game):
+ if game.snake[0][0]==10:
+ game.direction=6
+
+for i in range(0,10):
+ score=game.run()
+ print("Game ended with "+str(score)) \ No newline at end of file