from baselines.common.atari_wrappers import make_atari, wrap_deepmind
from dqn_agent import Agent
pong_game = "PongNoFrameskip-v4"
pong_reward_threshold = 19.0
pong_model_filepath = "pong_model.json"
pong_model_weights_filepath = "pong_model.h5"
env = make_atari(pong_game)
env = wrap_deepmind(env, frame_stack=True, scale=True)
env.seed(seed)
pong_agent = Agent(env, reward_threshold=pong_reward_threshold,
model_filepath=pong_model_filepath,
model_weights_filepath=pong_model_weights_filepath)
pong_agent.train()
pong_agent.run()
from baselines.common.atari_wrappers import make_atari, wrap_deepmind
from dqn_agent import Agent
seed = 42
env = make_atari("BreakoutNoFrameskip-v4")
# Warp the frames, grey scale, stake four frame and scale to smaller ratio
env = wrap_deepmind(env, frame_stack=True, scale=True)
env.seed(seed)
breakout_agent = Agent(env)
breakout_agent.train()
breakout_agent.run()
