sccross.models.utils.ZILN

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

Bases: torch.distributions.log_normal.LogNormal

Zero-inflated log-normal distribution with subsetting support

Parameters

  • zi_logits (Tensor) – Zero-inflation logits

  • loc (Tensor) – Location of the log-normal distribution

  • scale (Tensor) – Scale of the log-normal distribution

Methods

log_prob

Scores the sample by inverting the transform(s) and computing the score using the score of the base distribution and the log abs det jacobian.

Attributes