def main(args):
rng = Random(args.seed)
network_parameters = (105, 103, 100)
trainer = Trainer(
battleship_fitness,
args.class_size,
network_parameters,
args.cream,
rng,
args.threads
)
if args.class_location:
_, _, files = next(os.walk(args.class_location))
networks = [os.path.join(args.class_location, f)
for f in files if f.endswith(".network")]
trainer.stable = [Network.load_from_file(f) for f in networks]
else:
trainer.spawn_class()
for n in range(args.rounds):
print "Beginning round", n
trainer.run_round(battleship_trainer)
if args.output_location:
generation_dir = os.path.join(
args.output_location,
"gen_{}".format(max(n.generation for n in trainer.stable))
)
os.makedirs(generation_dir)
for network in trainer.stable:
network.save_to_file(path=generation_dir)
network, score = trainer.best
network.save_to_file(filename="best_gen{}_{}.network".format(
network.generation, score), path=args.output_location)
