lit_sb3.distributions.maskable

source package lit_sb3.distributions.maskable

Modules

• lit_sb3.distributions.maskable.base — Base class for maskable distributions.

• lit_sb3.distributions.maskable.bernoulli — Maskable Bernoulli distribution for multi-binary action spaces.

• lit_sb3.distributions.maskable.categorical — Maskable categorical distribution for discrete action spaces.

• lit_sb3.distributions.maskable.maskable_categorical — Modified PyTorch Categorical distribution with action masking support.

• lit_sb3.distributions.maskable.multi_categorical — Maskable multi-categorical distribution for multi-discrete action spaces.