Ejemplos de AtariRLDqn en Python

Ejemplo n.º 1

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()

Ejemplo n.º 2

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) + "_")

Ejemplo n.º 3

Archivo: agent_test.py Proyecto: michalnand/rl_examples

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()

Ejemplo n.º 4

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()

Ejemplo n.º 5

Archivo: atari_enduro.py Proyecto: michalnand/rl_examples

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()