Distributions

class cellarium.ml.distributions.NegativeBinomial(mu: Tensor, theta: Tensor, validate_args: bool | None = None)[source]

Bases: TorchDistribution

Negative binomial distribution.

Parameters:

  • mu (Tensor) – Mean of the distribution.

  • theta (Tensor) – Inverse dispersion.

  • validate_args (bool | None)