sccross.models.utils.ZIN

class sccross.models.utils.ZIN(zi_logits, loc, scale)[source]

Bases: torch.distributions.normal.Normal

Zero-inflated normal distribution with subsetting support

Parameters

  • zi_logits (Tensor) – Zero-inflation logits

  • loc (Tensor) – Location of the normal distribution

  • scale (Tensor) – Scale of the normal distribution

Methods

log_prob

Returns the log of the probability density/mass function evaluated at value.

Attributes