[rllib] Behavior Cloning (#1400)

* Behavior Cloning

* episode_reward_mean -> mean_loss

* removing vestigial code

* punctuation

* unnecessary

* Behavior Cloning

* Behavior Cloning

* Update __init__.py
This commit is contained in:
Roy Fox
2018-01-23 10:50:45 -08:00
committed by Eric Liang
parent ee36effd8e
commit 4b0ef5eb2c
11 changed files with 390 additions and 83 deletions
+5 -2
View File
@@ -128,7 +128,7 @@ class Agent(Trainable):
self._initialize_ok = True
def _init(self, config, env_creator):
def _init(self):
"""Subclasses should override this for custom initialization."""
raise NotImplementedError
@@ -293,7 +293,7 @@ class Agent(Trainable):
raise NotImplementedError
def _restore(self):
def _restore(self, checkpoint_path):
"""Subclasses should override this to implement restore()."""
raise NotImplementedError
@@ -371,6 +371,9 @@ def get_agent_class(alg):
elif alg == "A3C":
from ray.rllib import a3c
return a3c.A3CAgent
elif alg == "BC":
from ray.rllib import bc
return bc.BCAgent
elif alg == "script":
from ray.tune import script_runner
return script_runner.ScriptRunner