def run(args):
# not using atari_wrapper
if args.train_pg:
env_name = args.env_name or 'Pong-v0'
env = Environment(env_name, args)
from agent_dir.agent_pg import Agent_PG
agent = Agent_PG(env, args)
agent.train()
# using atari_wrapper
if args.train_dqn:
env_name = args.env_name or 'BreakoutNoFrameskip-v4'
env = Environment(env_name, args, atari_wrapper=True)
from agent_dir.agent_dqn import Agent_DQN
agent = Agent_DQN(env, args)
agent.train()
if args.test_pg:
env = Environment('Pong-v0', args, test=True)
from agent_dir.agent_pg import Agent_PG
agent = Agent_PG(env, args)
test(agent, env)
if args.test_dqn:
env = Environment('BreakoutNoFrameskip-v4',
args,
atari_wrapper=True,
test=True)
from agent_dir.agent_dqn import Agent_DQN
agent = Agent_DQN(env, args)
test(agent, env, total_episodes=100)
def run(args):
if args.test_dqn:
env = Environment('SeaquestNoFrameskip-v0',
args,
atari_wrapper=True,
test=True)
from agent_dir.agent_dqn import Agent_DQN
agent = Agent_DQN(env, args)
play_game(agent, env, total_episodes=1)
def run(args):
if args.test_dqn:
env = Environment('BreakoutNoFrameskip-v4',
args,
atari_wrapper=True,
test=True)
from agent_dir.agent_dqn import Agent_DQN
agent = Agent_DQN(env, args)
test(agent, env, total_episodes=100)
def run(args):
if args.test_pg:
env = gym.make('Pong-v0')
from agent_dir.agent_pg import Agent_PG
agent = Agent_PG(env, args)
test(agent, env)
if args.test_dqn:
env = make_wrap_atari(args.env_id, clip_rewards=False)
from agent_dir.agent_dqn import Agent_DQN
agent = Agent_DQN(args, env)
test(agent, env, total_episodes=100)
def run(args):
# All frames are preprocessed with atari wrapper.
if args.train_dqn:
env_name = args.env_name or 'SeaquestNoFrameskip-v0' #'BreakoutNoFrameskip-v4'
env = Environment(env_name, args, atari_wrapper=True)
from agent_dir.agent_dqn import Agent_DQN
agent = Agent_DQN(env, args)
agent.train()
if args.test_dqn:
env_name = args.env_name or 'SeaquestNoFrameskip-v0'
env = Environment(env_name,
args,
atari_wrapper=True,
test=True,
frame_stack_and_origin=True)
from agent_dir.agent_dqn import Agent_DQN
agent = Agent_DQN(env, args)
if (args.visualize):
print("<< visualization >>\n")
play_game(args, agent, env, total_episodes=1)
else:
print("<< test >>\n")
test(agent, env)
def run(args):
if args.train:
from agent_dir.agent_dqn import Agent_DQN
env = make_wrap_atari(args.env_id, clip_rewards=True)
#env = gym.make('LunarLander-v2')
agent = Agent_DQN(args, env)
if args.n_steps > 1:
agent.nsteps_train()
else:
agent.train()
def run(args):
# All frames are preprocessed with atari wrapper.
if args.train:
env_name = args.env_name or 'BreakoutNoFrameskip-v4'
env = Environment(env_name, args, atari_wrapper=True)
agent = Agent_DQN(env, args)
agent.train()
if args.test:
env = Environment('BreakoutNoFrameskip-v4', args, atari_wrapper=True, test=True)
agent = Agent_DQN(env, args)
test(agent, env, total_episodes=arg.testEpisodes)
def run(args):
if args.train_dqn:
env_name = args.env_name or 'SpaceInvadersNoFrameskip-v4'
env = Environment(env_name, args, atari_wrapper=True)
from agent_dir.agent_dqn import Agent_DQN
agent = Agent_DQN(env, args)
agent.train()
if args.test_dqn:
env = Environment('SpaceInvadersNoFrameskip-v4', args, atari_wrapper=True, test=True)
from agent_dir.agent_dqn import Agent_DQN
agent = Agent_DQN(env, args)
test(agent, env, total_episodes=100)
def run(args):
if args.train_pg:
env_name = args.env_name or 'Pong-v0'
env = Environment(env_name, args)
from agent_dir.agent_pg import Agent_PG
agent = Agent_PG(env, args)
agent.train()
if args.train_pg_improved:
env_name = args.env_name or 'Pong-v0'
env = Environment(env_name, args)
from agent_dir.agent_pg_improved import Agent_PG
agent = Agent_PG(env, args)
agent.train()
if args.train_dqn:
env_name = args.env_name or 'BreakoutNoFrameskip-v4'
env = Environment(env_name, args, atari_wrapper=True)
from agent_dir.agent_dqn import Agent_DQN
agent = Agent_DQN(env, args)
agent.train()
if args.train_dqn_improved:
env_name = args.env_name or 'BreakoutNoFrameskip-v4'
env = Environment(env_name, args, atari_wrapper=True)
from agent_dir.agent_dqn_improved import Agent_DQN
agent = Agent_DQN(env, args)
agent.train()
if args.train_pong_ac:
env_name = args.env_name or 'Pong-v0'
env = Environment(env_name, args)
from agent_dir.agent_pong_ac import Agent_AC
agent = Agent_AC(env, args)
agent.train()
if args.train_break_ac:
env_name = args.env_name or 'BreakoutNoFrameskip-v4'
env = Environment(env_name, args, atari_wrapper=True)
from agent_dir.agent_break_ac import Agent_AC
agent = Agent_AC(env, args)
agent.train()
if args.train_pong_ac_improved:
env_name = args.env_name or 'Pong-v0'
env = Environment(env_name, args)
from agent_dir.agent_pong_ac_improved import Agent_AC
agent = Agent_AC(env, args)
agent.train()
if args.train_break_ac_improved:
env_name = args.env_name or 'BreakoutNoFrameskip-v4'
env = Environment(env_name, args, atari_wrapper=True)
from agent_dir.agent_break_ac_improved import Agent_AC
agent = Agent_AC(env, args)
agent.train()
if args.test_pg:
env = Environment('Pong-v0', args, test=True)
from agent_dir.agent_pg import Agent_PG
agent = Agent_PG(env, args)
test(agent, env)
if args.test_pg_improved:
env = Environment('Pong-v0', args, test=True)
from agent_dir.agent_pg_improved import Agent_PG
agent = Agent_PG(env, args)
test(agent, env)
if args.test_dqn:
env = Environment('BreakoutNoFrameskip-v4',
args,
atari_wrapper=True,
test=True)
from agent_dir.agent_dqn import Agent_DQN
agent = Agent_DQN(env, args)
test(agent, env, total_episodes=100)
if args.test_dqn_improved:
env = Environment('BreakoutNoFrameskip-v4',
args,
atari_wrapper=True,
test=True)
from agent_dir.agent_dqn_improved import Agent_DQN
agent = Agent_DQN(env, args)
test(agent, env, total_episodes=100)
if args.train_ddqn:
env_name = args.env_name or 'BreakoutNoFrameskip-v4'
env = Environment(env_name, args, atari_wrapper=True)
from agent_dir.agent_ddqn import Agent_DQN
agent = Agent_DQN(env, args)
agent.train()
if args.test_ddqn:
env = Environment('BreakoutNoFrameskip-v4',
args,
atari_wrapper=True,
test=True)
from agent_dir.agent_ddqn import Agent_DQN
agent = Agent_DQN(env, args)
test(agent, env, total_episodes=100)
if args.train_dddqn:
env_name = args.env_name or 'BreakoutNoFrameskip-v4'
env = Environment(env_name, args, atari_wrapper=True)
from agent_dir.agent_dddqn import Agent_DQN
agent = Agent_DQN(env, args)
agent.train()
if args.test_dddqn:
env = Environment('BreakoutNoFrameskip-v4',
args,
atari_wrapper=True,
test=True)
from agent_dir.agent_dddqn import Agent_DQN
agent = Agent_DQN(env, args)
test(agent, env, total_episodes=100)
