wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-20 12:40:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

[rllib] Add more regression tests and autogenerate (#2324)

Browse Source
This commit is contained in:
Richard Liaw
2018-07-02 08:20:53 -07:00
committed by GitHub
parent 8aa56c12e6
commit f0ed1c1674
10 changed files with 55 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/rllib/tuned_examples/regression_tests/regression_test.py
-49
View File
@@ -51,52 +51,3 @@ class Regression():
def track_iterations(self, result):
return result["training_iteration"]
class TestCartPolePPO(Regression):
_file = "cartpole-ppo.yaml"
def setup_cache(self):
return _evaulate_config(self._file)
class TestCartPolePG(Regression):
_file = "cartpole-pg.yaml"
def setup_cache(self):
return _evaulate_config(self._file)
class TestPendulumDDPG(Regression):
_file = "pendulum-ddpg.yaml"
def setup_cache(self):
return _evaulate_config(self._file)
class TestCartPoleES(Regression):
_file = "cartpole-es.yaml"
def setup_cache(self):
return _evaulate_config(self._file)
class TestCartPoleDQN(Regression):
_file = "cartpole-dqn.yaml"
def setup_cache(self):
return _evaulate_config(self._file)
class TestCartPoleA3C(Regression):
_file = "cartpole-a3c.yaml"
def setup_cache(self):
return _evaulate_config(self._file)
class TestCartPoleA3CPyTorch(Regression):
_file = "cartpole-a3c-pytorch.yaml"
def setup_cache(self):
return _evaulate_config(self._file)