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()