def test_header(self): """Define the arena with one version""" arena = Arena([("Random A", lambda seed: AgentRandom(seed)), ("Random C", lambda seed: AgentRandom(seed)), ("Random B", lambda seed: AgentRandom(seed))], 5) self.assertListEqual(arena.csv_header(), ["opponent", "Random A", "Random B", "Random C"])
default='arena.results.csv', help='Path to write arena results') ARGS = PARSER.parse_args() print('Starting arena') A3C_PATH = os.path.join("models", "stated_2017-05-01T22-59-33.510476_best_0.45875") dtype = torch.cuda.FloatTensor if torch.cuda.is_available( ) else torch.FloatTensor ARENA = Arena( [ # Place agents in this list as created # first in the tuple is the readable name # second is a lambda that ONLY takes a random seed. This can be discarded # if the the Agent does not require a seed ("A3C", lambda seed: AgentA3C(A3C_PATH, dtype, seed)), ("Random", lambda seed: AgentRandom(seed)) ], 500) print('Run the arena for: ', ARENA.csv_header()) with open(ARGS.output, 'w') as f: WRITER = csv.writer(f) WRITER.writerow(ARENA.csv_header()) WRITER.writerows(ARENA.csv_results_lists()) print('Complete')