Fixed graph features

- Common graph code made into a method.
- No longer needs the user to pass in the ga for graphing.
- Changed graph attribute from class to object.
- Added ga to the graph object as an attribute on initialization to avoid needing to pass it in every time you graph.
- Capitalized database/graph classes.
This commit is contained in:
SimpleArt
2020-11-07 12:43:47 -05:00
parent 87beba1209
commit f1f9d70c26
5 changed files with 47 additions and 42 deletions

View File

@ -50,7 +50,7 @@ class GA(Attributes):
# Create the database here to allow the user to change
# the database name and structure in the running function.
self.database = database.database()
self.database = database.Database()
self.database.create_data_table(self)
# Create the initial population
@ -158,3 +158,9 @@ class GA(Attributes):
"""Prints the best chromosome and its fitness"""
print(f"Best Chromosome \t: {self.population.get_chromosome(0)}")
print(f"Best Fitness \t: {self.population.get_chromosome(0).get_fitness()}")
def print_worst(self):
"""Prints the worst chromosome and its fitness"""
print(f"Worst Chromosome \t: {self.population.get_chromosome(-1)}")
print(f"Worst Fitness \t: {self.population.get_chromosome(-1).get_fitness()}")