scglue.models.prob.ZIN

class scglue.models.prob.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