default=100000)
# GAME options
parser.add_argument("--n_actions",
type=int,
help="number of game output actions",
default=2)
parser.add_argument("--frame_size",
type=str,
help="size of game frame in pixels",
default=84)
if __name__ == '__main__':
options = parser.parse_args()
# Select agent
if options.algo == 'dqn':
agent = DQNAgent(options)
elif options.algo == 'a2c':
agent = A2CAgent(options)
elif options.algo == 'ppo':
agent = PPOAgent(options)
else:
print("ERROR. This algorithm has not been implemented yet.")
# Train or evaluate agent
if options.mode == 'train':
agent.train()
elif options.mode == 'eval':
agent.play_game()
