garage.torch.policies.gaussian_mlp_policy module

GaussianMLPPolicy.

class GaussianMLPPolicy(env_spec, **kwargs)[source]

Bases: garage.torch.policies.base.Policy, garage.torch.modules.gaussian_mlp_module.GaussianMLPModule

GaussianMLPPolicy.

A policy that contains a MLP to make prediction based on a gaussian distribution.

Parameters:
  • env_spec (garage.envs.env_spec.EnvSpec) – Environment specification.
  • module – GaussianMLPModule to make prediction based on a gaussian
  • distribution. –
Returns:
forward(inputs)[source]

Forward method.

get_action(observation)[source]

Get a single action given an observation.

get_actions(observations)[source]

Get actions given observations.

get_entropy(observation)[source]

Get entropy given observations.

log_likelihood(observation, action)[source]

Get log likelihood given observations and action.

reset(dones=None)[source]

Reset the environment.

vectorized

Vectorized or not.