s444409/DSZI_Survival
s444409/DSZI_Survival
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added file logging

Browse Source
This commit is contained in:
Marcin Kostrzewski 2020-05-17 00:38:36 +02:00
parent 7c6cc737e1
commit 0225eda1eb
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/AI/GA.py
View File

@ -1,4 +1,6 @@
import random import random
from datetime import datetime
import numpy import numpy
import copy import copy
@ -32,6 +34,11 @@ def geneticAlgorithm(map, iter, solutions, mutationAmount=0.05, multithread=Fals
for i in range(solutions): for i in range(solutions):
maps.append(Map(map.filename, map.screen)) maps.append(Map(map.filename, map.screen))
# Initialize log file
with open("results.txt", "w+") as f:
f.write("GA Results from " + str(datetime.now()))
f.write("\n")
for i in range(iter): for i in range(iter):
print("\nRunning {} generation...".format(i + 1)) print("\nRunning {} generation...".format(i + 1))
fitness = [] fitness = []
@ -47,12 +54,21 @@ def geneticAlgorithm(map, iter, solutions, mutationAmount=0.05, multithread=Fals
for t in threads: for t in threads:
t.join() t.join()
fitness.append(t.getResult()) fitness.append(t.getResult())
parents = selectMatingPool(population, fitness, int(solutions / 2)) parents = selectMatingPool(population, fitness, int(solutions / 2))
print("Best fitness: {}".format(max(fitness))) print("Best fitness: {}".format(max(fitness)))
offspring = mating(parents, solutions, mutationAmount) offspring = mating(parents, solutions, mutationAmount)
print("Best offspring: ", offspring[0]) print("Best offspring: ", offspring[0])
population = offspring population = offspring
# Add info to logfile
with open("results.txt", "a") as f:
f.write("Population: {}\n".format(i))
f.write("Best fitness: {}\n".format(fitness[0]))
f.write("Best offspring: " + str(offspring[0]))
f.write("\n\n")
def selectMatingPool(population, fitness, count): def selectMatingPool(population, fitness, count):
""" """