Rewrote data_list to __iter__

This commit is contained in:
SimpleArt
2020-11-19 10:17:22 -05:00
parent 857e248034
commit 6e95ff5d9d
3 changed files with 25 additions and 22 deletions

View File

@ -71,7 +71,7 @@ class Crossover_Methods:
return ga.make_chromosome([ # Make a new chromosome
ga.make_gene(random.randint(*sorted([data_1, data_2]))) # by randomly selecting integer genes between
for data_1, data_2 in zip(parent_1.data_list(), parent_2.data_list())]) # each parents' genes
for data_1, data_2 in zip(iter(parent_1), iter(parent_2))]) # each parents' genes
def int_weighted(ga, parent_1, parent_2):
@ -84,7 +84,7 @@ class Crossover_Methods:
ga.make_gene(int( # filled with new integer genes
weight*data_1+(1-weight)*data_2 # with weight% from gene 1 and
)) # (100-weight)% from gene 2
for data_1, data_2 in zip(parent_1.data_list(), parent_2.data_list())]) # from each parents' genes
for data_1, data_2 in zip(iter(parent_1), iter(parent_2))]) # from each parents' genes
def float_random(ga, parent_one, parent_two):
@ -92,7 +92,7 @@ class Crossover_Methods:
return ga.make_chromosome([ # Make a new chromosome
ga.make_gene(random.uniform([data_1, data_2])) # by randomly selecting integer genes between
for data_1, data_2 in zip(parent_1.data_list(), parent_2.data_list())]) # from each parents' genes
for data_1, data_2 in zip(iter(parent_1), iter(parent_2))]) # from each parents' genes
def float_weighted(ga, parent_one, parent_two):
@ -105,4 +105,4 @@ class Crossover_Methods:
ga.make_gene( # filled with new float genes
weight*data_1+(1-weight)*data_2 # with weight% from gene 1 and
) # (100-weight)% from gene 2
for data_1, data_2 in zip(parent_1.data_list(), parent_2.data_list())]) # from each parents' genes
for data_1, data_2 in zip(iter(parent_1), iter(parent_2))]) # from each parents' genes

View File

@ -53,9 +53,11 @@ class Chromosome:
"""Set the fitness value of the chromosome"""
self.fitness = fitness
def data_list(self):
"""Returns a list of the gene values"""
return [gene.value for gene in self.gene_list]
def __iter__(self):
"""Returns an iterable of the gene values"""
for gene in self.gene_list:
yeild gene.value
def __repr__(self):

View File

@ -149,9 +149,10 @@ class Population:
self.fitness = fitness
def data_list(self):
"""Returns a list of chromosome data lists"""
return [chromosome.data_list() for chromosome in self.chromosome_list]
def __iter__(self):
"""Returns an iterable of chromosome iterables"""
for chromosome in self.chromosome_list:
yield iter(chromosome)
def __repr__(self):