paulin/EasyGA
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Improved the sort by fitness function

Browse Source
This commit is contained in:
SimpleArt
2020-10-12 22:43:41 -04:00
parent f277bc1684
commit 86fbd1a9c0
1 changed files with 8 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/EasyGA.py
View File

@ -76,6 +76,7 @@ class GA:
self.set_all_fitness(self.population.chromosome_list)
self.population.set_all_chromosomes(self.sort_by_best_fitness(self.population.get_all_chromosomes()))
else:
self.set_all_fitness(self.population.chromosome_list)
self.parent_selection_impl(self)
next_population = self.crossover_population_impl(self)
next_population = self.survivor_selection_impl(self, next_population)
@ -123,17 +124,10 @@ class GA:
def sort_by_best_fitness(self, chromosome_set):
"""Sorts the array by fitness.
1st element has highest fitness.
2nd element has second highest fitness.
etc.
"""
chromosome_set_temp = chromosome_set
not_sorted_check = 0
while (not_sorted_check != len(chromosome_set_temp)):
not_sorted_check = 0
for i in range(len(chromosome_set_temp)):
if ((i + 1 < len(chromosome_set_temp)) and (chromosome_set_temp[i + 1].fitness > chromosome_set_temp[i].fitness)):
chromosome_set[i], chromosome_set_temp[i + 1] = chromosome_set_temp[i + 1], chromosome_set[i]
else:
not_sorted_check += 1
chromosome_set = chromosome_set_temp
return chromosome_set
return list(reversed([chromosome for chromosome in sorted(chromosome_set, key = lambda chromo: chromo.get_fitness())]))