Bernoulli

distreqx.distributions.Bernoulli(distreqx.distributions.AbstractSampleLogProbDistribution, distreqx.distributions.AbstractSTDDistribution, distreqx.distributions.AbstractSurvivalDistribution)

Bernoulli distribution over \(\{0, 1\}\).

For success probability \(p = P(X = 1)\), the probability mass function is

\[ P(X = x) = p^x (1 - p)^{1 - x}, \quad x \in \{0, 1\}. \]

__init__(logits: float | Array | None = None, probs: float | Array | None = None)

Initializes a Bernoulli distribution.

Arguments:

• logits: Log-odds \(\ell = \log(p / (1 - p))\) for the event \(X = 1\). Only one of logits or probs can be specified.
• probs: Probability \(p = P(X = 1)\). Only one of logits or probs can be specified.

---