diff options
Diffstat (limited to 'qlearning.py')
| -rwxr-xr-x | qlearning.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/qlearning.py b/qlearning.py index 5cb3d8a..a1770b6 100755 --- a/qlearning.py +++ b/qlearning.py @@ -38,7 +38,6 @@ def isWall(h,game): return(True) return(False) - def event_handler(game,event): h=game.snake[0] left=(h[0]-1,h[1]) @@ -62,6 +61,14 @@ def event_handler(game,event): obstacle_down=(down in game.snake or isWall(down, game)) obstacle_left=(left in game.snake or isWall(left, game)) + reward=0 + if event==1: + reward=1 + elif event==-1: + reward=-10 + + state=2048*snake_go_up + if game.snake[0][0]==10: game.direction=6 |