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-02 09:29:45 +01:00
parent 5b3efc4a21
commit 37308a0d4c
3 changed files with 8207 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
qlearning.py
View file

@ -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

8192
qtable.txt Normal file
View file

File diff suppressed because it is too large Load diff

19
snake.py
View file

@ -129,7 +129,7 @@ class Snake:
self.draw_infos()
pygame.display.flip()
def play(self,direction):
def play(self,direction,handle_quit=True):
"""
Play using wall clock directions (12=up, 3=right, 6=down and 9=left)
"""
@ -151,10 +151,11 @@ class Snake:
self.draw()
self.clock.tick(self.fps)
# Ensure we not quit
for event in pygame.event.get():
if event.type == pygame.QUIT:
pygame.quit()
sys.exit()
if handle_quit:
for event in pygame.event.get():
if event.type == pygame.QUIT:
pygame.quit()
sys.exit()
return(code)
def play2(self,direction):
@ -190,7 +191,10 @@ class Snake:
while True:
# Check inputs
for event in pygame.event.get():
if event.type == pygame.KEYDOWN:
if event.type == pygame.QUIT:
pygame.quit()
sys.exit()
elif event.type == pygame.KEYDOWN:
if event.key == pygame.K_LEFT and self.direction != 3:
self.direction=9
break
@ -204,5 +208,6 @@ class Snake:
self.direction=6
break
if self.play(self.direction) <0:
if self.play(self.direction,handle_quit=False) <0:
break