def run(network_path, agent_type): envs = [] size = 48 envs.append(libs.libs_env.env_atari_arkanoid.EnvAtariArkanoid(size)) envs.append(libs.libs_env.env_atari_enduro.EnvAtariEnduro(size)) envs.append(libs.libs_env.env_atari_invaders.EnvAtariInvaders(size)) envs.append(libs.libs_env.env_atari_pacman.EnvAtariPacman(size)) envs.append(libs.libs_env.env_atari_pong.EnvAtariPong(size)) envs.append(libs.libs_env.env_atari_snake.EnvAtariSnake(size)) #init environment env = libs.libs_env.env_atari_multi.EnvAtariMulti(envs, size, 512) rl_atari = atari_rl_dqn.AtariRLDqn(env, network_path, agent_type) rl_atari.visualise()
def test(env_id, network_path): size = 48 if env_id == 0: env = libs.libs_env.env_atari_arkanoid.EnvAtariArkanoid(size) elif env_id == 1: env = libs.libs_env.env_atari_enduro.EnvAtariEnduro(size) elif env_id == 2: env = libs.libs_env.env_atari_invaders.EnvAtariInvaders(size) elif env_id == 3: env = libs.libs_env.env_atari_pacman.EnvAtariPacman(size) elif env_id == 4: env = libs.libs_env.env_atari_pong.EnvAtariPong(size) elif env_id == 5: env = libs.libs_env.env_atari_snake.EnvAtariSnake(size) rl_atari = atari_rl_dqn.AtariRLDqn(env, network_path) rl_atari.test(str(env_id) + "_")
import libs.libs_env.env_pong import libs.libs_agent.agent #init environment #env = libs.libs_env.env_mountain_car.EnvMountainCar() #env = libs.libs_env.env_atari_arkanoid.EnvAtariArkanoid(24) env = libs.libs_env.env_pong.EnvPong(9) #env = libs.libs_env.env_atari_snake.EnvAtariSnake(24) #env = libs.libs_env.env_mountain_car.EnvMountainCar() ''' network_path = "networks/dqn/" rl_atari = atari_rl_dqn.AtariRLDqn(env, network_path, "dqn") rl_atari.train(1000) #rl_atari.visualise() ''' ''' network_path = "networks/dqn_dueling/" rl_atari = atari_rl_dqn.AtariRLDqn(env, network_path, "dqn_dueling") rl_atari.train(200) rl_atari.visualise() ''' network_path = "networks/actor_critic/" rl_atari = atari_rl_dqn.AtariRLDqn(env, network_path, "actor_critic") rl_atari.train(1000) #rl_atari.visualise()
import libs.libs_env.env_atari_pacman import atari_rl_dqn #init environment env = libs.libs_env.env_atari_pacman.EnvAtariPacman(48) network_path = "networks/atari/pacman/" rl_atari = atari_rl_dqn.AtariRLDqn(env, network_path) ''' network_path = "networks/atari/pacman_dueling_dqn/" rl_atari = atari_rl_dqn.AtariRLDqn(env, network_path, "dqn_dueling") ''' #rl_atari.train(1000) rl_atari.visualise()
import libs.libs_env.env_atari_enduro import atari_rl_dqn #init environment env = libs.libs_env.env_atari_enduro.EnvAtariEnduro(48) ''' network_path = "networks/atari/enduro/" rl_atari = atari_rl_dqn.AtariRLDqn(env, network_path) ''' network_path = "networks/atari/enduro_dueling_dqn/" rl_atari = atari_rl_dqn.AtariRLDqn(env, network_path, "dqn_dueling") #rl_atari.train(1000) rl_atari.visualise()