scglue.models.prob.ZILN

class scglue.models.prob.ZILN(zi_logits, loc, scale)

Bases: LogNormal

Zero-inflated log-normal distribution with subsetting support

Parameters:

• zi_logits (torch.Tensor) – Zero-inflation logits

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

• scale (torch.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