loic/snakeq
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Minor changes

Browse source
This commit is contained in:
Loic Guegan 2022-11-01 17:20:36 +01:00
parent e386840172
commit 69c5d709a4
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
qlearning.py
View file

@ -1,5 +1,5 @@
#!/usr/bin/env python #!/usr/bin/env python
import sys import sys,random
import numpy as np import numpy as np
# Import snake game # Import snake game
@ -69,9 +69,14 @@ def event_handler(game,event):
# This come from me I do not now if it is the best way to identify a state # This come from me I do not now if it is the best way to identify a state
state=2**11*snake_go_up+2**10*snake_go_right+2**9*snake_go_down+2**8*snake_go_left+2**7*apple_up+2**6*apple_right+2**5*apple_down+2**4*apple_left+2**3*obstacle_up+2**2*obstacle_right+2**1*obstacle_down+obstacle_left state=2**11*snake_go_up+2**10*snake_go_right+2**9*snake_go_down+2**8*snake_go_left+2**7*apple_up+2**6*apple_right+2**5*apple_down+2**4*apple_left+2**3*obstacle_up+2**2*obstacle_right+2**1*obstacle_down+obstacle_left
if game.snake[0][0]==10:
game.direction=6 if np.max(qtable[state]) > 0:
action = np.argmax(qtable[state])
else:
action=random.choice((12,3,6,9))
game.direction=action
for i in range(0,10): for i in range(0,10):
score=game.run(event_handler=event_handler) score=game.run(event_handler=event_handler)