class Trainer():
def __init__(self, withGrapics=True):
self.bestbest = 0
self.done = False
self.pool = Pool(Config.WIDTH, Config.HEIGHT, self.callback)
self.graphics = Graphics()
if withGrapics:
self.graphics.init("PoolGame", Config.SIZE)
def callback(self, gen, avg_score, best_score, best_json):
line = "Generation\t%d\tAverage Score\t%f\tBest Score\t%f" % (
gen, avg_score, best_score)
self.write_log(Config.JSON_FOLDER + "/" + "logfile.txt", line)
filename = "brain-g%03d-%04d.json" % (gen, best_score * 1000)
if best_score >= self.bestbest:
if best_json != None:
self.write_file(Config.JSON_FOLDER + "/" + filename, best_json)
self.bestbest = best_score
if gen == Config.GENERATIONS:
self.done = True
def write_log(self, filename, line):
print(line)
with open(filename, "a") as outfile:
outfile.write(line + "\n")
def write_file(self, filename, data):
with open(filename, "w") as outfile:
outfile.write(data + "\n")
def run(self):
# Loop until the user clicks the close button.
while not self.done:
self.done = self.graphics.queryQuit()
# Set the screen background
self.graphics.fill(Config.GREEN)
self.graphics.print("Clock: {}".format(self.graphics.fps()))
# Do physics
self.pool.tick()
# Draw everything
self.pool.draw(self.graphics)
# Update screen
self.graphics.flip()
# Exit
self.graphics.quit()
def callback(gen, avg_score, best_score, best_json): global bestbest global done line = "Generation\t%d\tAverage Score\t%f\tBest Score\t%f" % (gen, avg_score, best_score) write_log(Config.JSON_FOLDER + "/" + "logfile.txt", line) filename = "brain-g%03d-%04d.json" % (gen, best_score * 1000) if best_score >= bestbest: if best_json != None: write_file(Config.JSON_FOLDER + "/" + filename, best_json) bestbest = best_score if gen == Config.GENERATIONS: done = True def write_log(filename, line): print(line) with open(filename, "a") as outfile: outfile.write(line + "\n") def write_file(filename, data): with open(filename, "w") as outfile: outfile.write(data + "\n") pool = Pool(Config.WIDTH, Config.HEIGHT, callback) while done == False: pool.tick()
