In [ ]:
import gym
from environment import Environment
from train import Trainer
from dqn import DQN
env = Environment(args)
agent = DQN (env,args)
#Train Agent
Trainer(agent).run()
#Play the game
env.gym.monitor.start(args.out,force=True)
agent.play()
env.gym.monitor.close()