ray.rllib.core.rl_module.rl_module.RLModule.get_exploration_action_dist_cls

RLModule.get_exploration_action_dist_cls() → Type[ray.rllib.models.distributions.Distribution]

Returns the action distribution class for this RLModule used for exploration.

This class is used to create action distributions from outputs of the forward_exploration method. If the case that no action distribution class is needed, this method can return None.

Note that RLlib’s distribution classes all implement the Distribution interface. This requires two special methods: Distribution.from_logits() and Distribution.to_deterministic(). See the documentation for Distribution for more detail.