Skip to content

lit_sb3.distributions

[docs] package lit_sb3.distributions 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
"""Probability distributions for reinforcement learning."""

from lit_sb3.distributions.base import Distribution, SelfDistribution, sum_independent_dims
from lit_sb3.distributions.bernoulli import BernoulliDistribution, SelfBernoulliDistribution
from lit_sb3.distributions.categorical import CategoricalDistribution, SelfCategoricalDistribution
from lit_sb3.distributions.gaussian import DiagGaussianDistribution, SelfDiagGaussianDistribution
from lit_sb3.distributions.multi_categorical import MultiCategoricalDistribution, SelfMultiCategoricalDistribution
from lit_sb3.distributions.sde import SelfStateDependentNoiseDistribution, StateDependentNoiseDistribution
from lit_sb3.distributions.squashed_gaussian import (
    SelfSquashedDiagGaussianDistribution,
    SquashedDiagGaussianDistribution,
)
from lit_sb3.distributions.tanh_bijector import TanhBijector

__all__ = [
    "Distribution",
    "SelfDistribution",
    "sum_independent_dims",
    "DiagGaussianDistribution",
    "SelfDiagGaussianDistribution",
    "SquashedDiagGaussianDistribution",
    "SelfSquashedDiagGaussianDistribution",
    "CategoricalDistribution",
    "SelfCategoricalDistribution",
    "MultiCategoricalDistribution",
    "SelfMultiCategoricalDistribution",
    "BernoulliDistribution",
    "SelfBernoulliDistribution",
    "StateDependentNoiseDistribution",
    "SelfStateDependentNoiseDistribution",
    "TanhBijector",
]