mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
69 lines
2.6 KiB
Python
69 lines
2.6 KiB
Python
import gym
|
|
import unittest
|
|
|
|
import ray
|
|
from ray.rllib.agents.dqn import DQNTrainer
|
|
from ray.rllib.agents.a3c import A3CTrainer
|
|
from ray.rllib.agents.dqn.dqn_tf_policy import _adjust_nstep
|
|
from ray.rllib.utils.test_utils import framework_iterator
|
|
from ray.tune.registry import register_env
|
|
|
|
|
|
class EvalTest(unittest.TestCase):
|
|
def test_dqn_n_step(self):
|
|
obs = [1, 2, 3, 4, 5, 6, 7]
|
|
actions = ["a", "b", "a", "a", "a", "b", "a"]
|
|
rewards = [10.0, 0.0, 100.0, 100.0, 100.0, 100.0, 100.0]
|
|
new_obs = [2, 3, 4, 5, 6, 7, 8]
|
|
dones = [0, 0, 0, 0, 0, 0, 1]
|
|
_adjust_nstep(3, 0.9, obs, actions, rewards, new_obs, dones)
|
|
self.assertEqual(obs, [1, 2, 3, 4, 5, 6, 7])
|
|
self.assertEqual(actions, ["a", "b", "a", "a", "a", "b", "a"])
|
|
self.assertEqual(new_obs, [4, 5, 6, 7, 8, 8, 8])
|
|
self.assertEqual(dones, [0, 0, 0, 0, 1, 1, 1])
|
|
self.assertEqual(rewards,
|
|
[91.0, 171.0, 271.0, 271.0, 271.0, 190.0, 100.0])
|
|
|
|
def test_evaluation_option(self):
|
|
def env_creator(env_config):
|
|
return gym.make("CartPole-v0")
|
|
|
|
agent_classes = [A3CTrainer, DQNTrainer]
|
|
|
|
for agent_cls in agent_classes:
|
|
for fw in framework_iterator(frameworks=("torch", "tf")):
|
|
ray.init(object_store_memory=1000 * 1024 * 1024)
|
|
register_env("CartPoleWrapped-v0", env_creator)
|
|
agent = agent_cls(
|
|
env="CartPoleWrapped-v0",
|
|
config={
|
|
"evaluation_interval": 2,
|
|
"evaluation_num_episodes": 2,
|
|
"evaluation_config": {
|
|
"gamma": 0.98,
|
|
"env_config": {
|
|
"fake_arg": True
|
|
}
|
|
},
|
|
"framework": fw,
|
|
})
|
|
# Given evaluation_interval=2, r0, r2, r4 should not contain
|
|
# evaluation metrics while r1, r3 should do.
|
|
r0 = agent.train()
|
|
r1 = agent.train()
|
|
r2 = agent.train()
|
|
r3 = agent.train()
|
|
|
|
self.assertTrue("evaluation" in r1)
|
|
self.assertTrue("evaluation" in r3)
|
|
self.assertFalse("evaluation" in r0)
|
|
self.assertFalse("evaluation" in r2)
|
|
self.assertTrue("episode_reward_mean" in r1["evaluation"])
|
|
self.assertNotEqual(r1["evaluation"], r3["evaluation"])
|
|
ray.shutdown()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import pytest
|
|
import sys
|
|
sys.exit(pytest.main(["-v", __file__]))
|