lit_sb3.distributions.maskable.bernoulli

source module lit_sb3.distributions.maskable.bernoulli

Maskable Bernoulli distribution for multi-binary action spaces.

Classes

• MaskableBernoulliDistribution — Bernoulli distribution for multibinary actions. Supports invalid action masking.

source class MaskableBernoulliDistribution(action_dim: int)

Bases : MaskableMultiCategoricalDistribution

Bernoulli distribution for multibinary actions. Supports invalid action masking.

Parameters

• action_dim : int — Number of binary actions