sNNakeCode/trainGeneticAlgorithm.py at trunk · jackdavidweber/sNNakeCode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
#*********************************************************************************
#trainGeneticAlgorithm.py
#Author: Craig Haber
#5/9/2020
#This program was used to train a genetic algorithm in order to create
#the intelligent Snake Game agents that can be observed in test_trained_agents.py
#For more detailed information, check out:
#https://craighaber.github.io/AI-for-Snake-Game/
#*********************************************************************************
#Instructions:
#Simply run the module to observe how the genetic algorithm was trained in action,
#starting from randomized chromosomes.
#Specific information about each population is saved in Gadata.txt in the same
#folder as this program, and the file will be created if it does not already exist
#after the first generation.
#Also, for every 10 populations, the population is saved in a file
#in the populations directory.
#*********************************************************************************
#Dependecies:
#
#To run this module, you must have the module pygame installed.
#Type pip install pygame in the command prompt or terminal to install it.
#If necessary, more specific instructions for installing pygame are here:
#https://www.pygame.org/wiki/GettingStarted
#
#Also, a Python version of 3.7 or higher is required.
#*********************************************************************************
import pygame
import argparse
from helpers.snakeGameGATrain import SnakeGameGATrain
from helpers import geneticAlgorithm as ga
import os


def main(max_generations=200):
	"""Function to train the genetic algorithm for creating intelligent Snake Game agents."""
	game_fps = 3000
	chroms_per_gen = 200
	num_inputs = 9
	num_hidden_layer_nodes = 10
	bits_per_weight = 8
	num_outputs = 4
	total_bits = ((num_inputs+1)*num_hidden_layer_nodes + num_hidden_layer_nodes*(num_hidden_layer_nodes+1) + num_outputs*(num_hidden_layer_nodes + 1))*bits_per_weight
	population = ga.genPopulation(chroms_per_gen, total_bits)
	game = SnakeGameGATrain(game_fps, population, chroms_per_gen, bits_per_weight, num_inputs, num_hidden_layer_nodes, num_outputs)

	pygame.font.init()

	while game.play and game.num_generations < max_generations:

		game.clock.tick(game.fps)

		game.move_snake()
		game.check_collisions()
		#check if snake is killed for not eating a fruit in a while
		game.update_frames_since_last_fruit()
		game.frames_alive += 1

		if game.restart == True:
			game.restart = False
			continue

		game.redraw_window()

		game.event_handler()


if __name__ == '__main__':
	# Set options to activate or deactivate the game view, and its speed
	parser = argparse.ArgumentParser()
	parser.add_argument("--max_generations", nargs='?', type=int, default=200)
	parser.add_argument("--headless", nargs='?', type=bool, default=False)
	args = parser.parse_args()
	print("Args", args)

	# If being run on a headless server, set driver to dummy
	# https://www.pygame.org/wiki/HeadlessNoWindowsNeeded
	if args.headless:
		os.environ["SDL_VIDEODRIVER"] = "dummy"

	main(max_generations=args.max_generations)