f1f9d70c26
- 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.
96 lines
2.6 KiB
Python
96 lines
2.6 KiB
Python
# Graphing package
|
|
import matplotlib.pyplot as plt
|
|
# Database class
|
|
from database import database
|
|
from sqlite3 import Error
|
|
|
|
class Graph:
|
|
"""Prebuilt graphing functions to make visual represention of fitness data."""
|
|
|
|
|
|
def __init__(self, ga):
|
|
self.ga = ga
|
|
|
|
|
|
def make_plot(self, type_of_plot, size, X, Y):
|
|
"""Create the plot"""
|
|
|
|
# Set the plot size
|
|
plt.figure(figsize = size)
|
|
|
|
if(type_of_plot == "line"):
|
|
plt.plot(X, Y)
|
|
elif(type_of_plot == "scatter"):
|
|
plt.scatter(X, Y)
|
|
elif(type_of_plot == "bar"):
|
|
plt.bar(X, Y)
|
|
|
|
|
|
def generation_total_fitness(self, type_of_plot = "line", size = [6,6]):
|
|
"""Show a plot of generation by generation total fitness."""
|
|
|
|
# Query the X data
|
|
generations = self.ga.database.get_total_generations()
|
|
|
|
# Create the generations list - [0,1,2,etc]
|
|
X = list(range(0, generations))
|
|
|
|
# Query for Y data
|
|
Y = self.ga.database.get_generation_total_fitness()
|
|
|
|
self.make_plot(type_of_plot, size, X, Y)
|
|
|
|
# x and y labels
|
|
plt.xlabel('Generation')
|
|
plt.ylabel('Generation Total Fitness')
|
|
plt.title('Relationship Between Generations and Generation Total Fitness')
|
|
|
|
# Show the plot
|
|
plt.show()
|
|
|
|
|
|
def highest_value_chromosome(self, type_of_plot = "line", size = [6,6]):
|
|
"""Generation by Max value chromosome """
|
|
|
|
# Query the X data
|
|
generations = self.ga.database.get_total_generations()
|
|
|
|
# Create the generations list - [0,1,2,etc]
|
|
X = list(range(0, generations))
|
|
|
|
# Query for Y data
|
|
Y = self.ga.database.get_highest_chromosome()
|
|
|
|
self.make_plot(type_of_plot, size, X, Y)
|
|
|
|
# x and y labels
|
|
plt.xlabel('Generation')
|
|
plt.ylabel('Generation Highest Fitness Chromosome')
|
|
plt.title('Relationship Between Generations and Highest Value Chromosome')
|
|
|
|
# Show the plot
|
|
plt.show()
|
|
|
|
|
|
def lowest_value_chromosome(self, type_of_plot = "line", size = [6,6]):
|
|
"""Generation by Min value Chromosome """
|
|
|
|
# Query the X data
|
|
generations = self.ga.database.get_total_generations()
|
|
|
|
# Create the generations list - [0,1,2,etc]
|
|
X = list(range(0, generations))
|
|
|
|
# Query for Y data
|
|
Y = self.ga.database.get_lowest_chromosome()
|
|
|
|
self.make_plot(type_of_plot, size, X, Y)
|
|
|
|
# x and y labels
|
|
plt.xlabel('Generation')
|
|
plt.ylabel('Generation Highest Fitness Chromosome')
|
|
plt.title('Relationship Between Generations and Lowest Value Chromosome')
|
|
|
|
# Show the plot
|
|
plt.show()
|