from snake_game import Snake
import time
import random
snake = Snake(20, True)
debug = False
best_score = 0
while 1:
time.sleep(0.03)
if(debug):
move = snake.do_move(0)
else:
move = snake.do_move(random.randint(0,4))
print(move)
if(move['reward'] < 0):
snake.reset()
elif(snake.get_score() > best_score):
best_score = snake.get_score()
snake = Snake(nb_cases, display)
grid_size = nb_cases + 2
agent = Trainer(allowed_actions=snake.get_actions_set(),
name='SnakeV1',
state_size=6)
reward = 0
bestScore = 0
iteration = 0
lastTotalReward = 0
while 1:
# 30 FPS
#time.sleep(0.033)
score = int(snake.get_score())
if (score > bestScore):
bestScore = score
print('New Best Score : ' + str(bestScore) + ' a ' +
str(datetime.datetime.now()))
agent.save()
if (reward < -1):
#print('Total reward : ' + str(lastTotalReward))
lastTotalReward = 0
snake.reset()
iteration += 1
observation = snake.get_observation()
