Calculate and return total fitness separately

This commit is contained in:
George Lacey 2017-09-23 19:09:38 +01:00
parent b4d9ea26bf
commit 8ea4b82c0d

View File

@ -9,8 +9,19 @@ class Population(object):
self.members.append(Individual()) self.members.append(Individual())
def fitness_function(self): def fitness_function(self):
for member in self.members:
member.fitness_function()
def total_fitness(self):
total = 0 total = 0
for member in self.members: for member in self.members:
total += member.fitness_function() total += member.fitness
return total return total
def roulette(self):
total = self.total_fitness()
for member in self.members:
total -= member.fitness
if total <= 0:
pass
# todo: add to crossover list