def __init__(self, environment=None, params=None):
BehaviorTFAAgent.__init__(self, environment=environment, params=params)
BehaviorContinuousML.__init__(self, params)
self._replay_buffer = self.GetReplayBuffer()
self._dataset = self.GetDataset()
self._collect_policy = self.GetCollectionPolicy()
self._eval_policy = self.GetEvalPolicy()
def __init__(self,
environment=None,
observer=None,
params=None,
init_gnn='init_interaction_network'):
"""
Initializes a `BehaviorGraphSACAgent` instance.
Args:
environment:
observer: The `GraphObserver` instance that generates the observations.
params: A `ParameterServer` instance containing parameters to configure
the agent.
"""
# the super init calls 'GetAgent', so assign the observer before
self._gnn_sac_params = params["ML"]["BehaviorGraphSACAgent"]
self._init_gnn = eval(init_gnn)
BehaviorTFAAgent.__init__(self,
environment=environment,
params=params,
observer=observer)
self._replay_buffer = self.GetReplayBuffer()
self._dataset = self.GetDataset()
self._collect_policy = self.GetCollectionPolicy()
self._eval_policy = self.GetEvalPolicy()
def __init__(self, environment=None, params=None, observer=None):
self._ppo_params = params["ML"]["BehaviorPPOAgent"]
BehaviorTFAAgent.__init__(self,
environment=environment,
params=params,
observer=observer)
self._replay_buffer = self.GetReplayBuffer()
self._collect_policy = self.GetCollectionPolicy()
self._eval_policy = self.GetEvalPolicy()
def __init__(self,
environment=None,
params=None,
observer=None):
self._sac_params = params["ML"]["BehaviorSACAgent"]
BehaviorTFAAgent.__init__(
self,
environment=environment,
params=params,
observer=observer)
BehaviorModel.__init__(self, params)
self._replay_buffer = self.GetReplayBuffer()
self._dataset = self.GetDataset()
self._collect_policy = self.GetCollectionPolicy()
self._eval_policy = self.GetEvalPolicy()
def __init__(self, environment=None, observer=None, params=None):
"""
Initializes a `BehaviorGraphSACAgent` instance.
Args:
environment:
observer: The `GraphObserver` instance that generates the observations.
params: A `ParameterServer` instance containing parameters to configure
the agent.
"""
# the super init calls 'GetAgent', so assign the observer before
self._observer = observer
BehaviorTFAAgent.__init__(self, environment=environment, params=params)
BehaviorContinuousML.__init__(self, params)
self._replay_buffer = self.GetReplayBuffer()
self._dataset = self.GetDataset()
self._collect_policy = self.GetCollectionPolicy()
self._eval_policy = self.GetEvalPolicy()