lit_sb3.utils.distribution

source module lit_sb3.utils.distribution

Probability distribution factory functions.

Functions

• make_proba_distribution — Return an instance of Distribution for the correct type of action space.

• make_masked_proba_distribution — Return an instance of Distribution for the correct type of action space.

source make_proba_distribution(action_space: spaces.Space, use_sde: bool = False, dist_kwargs: dict[str, Any] | None = None) → Distribution

Return an instance of Distribution for the correct type of action space.

Parameters

• action_space : spaces.Space — the input action space

• use_sde : bool — Force the use of StateDependentNoiseDistribution instead of DiagGaussianDistribution

• dist_kwargs : dict[str, Any] | None — Keyword arguments to pass to the probability distribution

Returns

• Distribution — the appropriate Distribution object

Raises

• NotImplementedError

source make_masked_proba_distribution(action_space: spaces.Space) → MaskableDistribution

Return an instance of Distribution for the correct type of action space.

Parameters

• action_space : spaces.Space — the input action space

Returns

• MaskableDistribution — the appropriate Distribution object

Raises

• NotImplementedError