Added more info printing

This commit is contained in:
Marcin Kostrzewski 2020-05-16 14:13:46 +02:00
parent 625b6956c9
commit e8e36a4420

View File

@ -21,11 +21,13 @@ def geneticAlgorithm(map, iter, solutions, mutationAmount=0.2):
initialPopulation = numpy.random.uniform(low=0.0, high=1.0, size=(solutions, weightsCount)) initialPopulation = numpy.random.uniform(low=0.0, high=1.0, size=(solutions, weightsCount))
population = initialPopulation population = initialPopulation
for i in range(iter): for i in range(iter):
print("Running {} generation...".format(i))
fitness = [] fitness = []
for player in population: for player in population:
fitness.append(doSimulation(player, map)) fitness.append(doSimulation(player, map))
parents = selectMatingPool(population, fitness, int(solutions / 2)) parents = selectMatingPool(population, fitness, int(solutions / 2))
print("Best fitness: {}".format(max(fitness)))
offspring = mating(parents, solutions, mutationAmount) offspring = mating(parents, solutions, mutationAmount)
# TODO: Parents selection, mating, offspring # TODO: Parents selection, mating, offspring
@ -46,7 +48,6 @@ def selectMatingPool(population, fitness, count):
bestIdxs.append(bestIdx) bestIdxs.append(bestIdx)
for id in bestIdxs: for id in bestIdxs:
result.append(population[id]) result.append(population[id])
print(result)
return result return result