Python DeepQNetwork.test 예제들

예제 #1

파일: main.py 프로젝트: mkimhi/TF2.0DQN-trajectory_buffer

def run_dqn(config, gym_wrapper, summaries_collector_traj,
            summaries_collector):
    q_network = DeepQNetwork(config, gym_wrapper, trajectory=1)
    initial_time = round(time(), 3)
    q_network.train(summaries_collector)
    reward = q_network.test(summaries_collector, episodes=10, render=True)
    summaries_collector.read_summaries('test')
    total_time_traj = round(time(), 3) - initial_time
    print("tested avg reward: {0} in: {1}".format(reward, total_time_traj))

예제 #2

from config_utils import read_main_config
from deep_q_network import DeepQNetwork
from gym_wrapper import GymWrapper


from tensorflow.python.framework.ops import disable_eager_execution
disable_eager_execution()

config = read_main_config()
gym_wrapper = GymWrapper(config['general']['scenario'])
deep_q_network = DeepQNetwork(config, gym_wrapper)
deep_q_network.train()
deep_q_network.test(episodes=3)