Bernoulli

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

Bernoulli distribution of shape dims.

Bernoulli distribution with parameter probs, the probability of outcome 1.

__init__(self, logits: Optional[Array] = None, probs: Optional[Array] = None)

Initializes a Bernoulli distribution.

Arguments:

• logits: Logit transform of the probability of a 1 event (0 otherwise), i.e. probs = sigmoid(logits). Only one of logits or probs can be specified.
• probs: Probability of a 1 event (0 otherwise). Only one of logits or probs can be specified.

---