def main():
args = parse_arguments()
agent_name = args.agent_name
filename = args.filename
num_epi = args.num_epi
print(agent_name, filename, num_epi)
pygame.init()
pygame.display.set_mode([1, 1])
task = MarioTask(initMarioMode=2)
if agent_name == 'human':
agent = HumanAgent(task.ACTION_MAPPING)
else:
agent = Forwardagent()
exp = EpisodicExperiment(task, agent)
print 'Task Ready'
exp.train(num_epi)
print 'mm 2:', task.reward
if agent_name == 'human':
print(all_action)
with open('./expert_data/' + filename + '_demo.pckl', 'wb') as f:
pickle.dump((exp.all_states, exp.all_actions), f)
print "finished"
pygame.quit()
