scglue.models.prob.ZIN
- class scglue.models.prob.ZIN(zi_logits, loc, scale)[source]
Bases:
Normal
Zero-inflated normal distribution with subsetting support
- Parameters:
zi_logits (
torch.Tensor
) – Zero-inflation logitsloc (
torch.Tensor
) – Location of the normal distributionscale (
torch.Tensor
) – Scale of the normal distribution
Methods
Returns the log of the probability density/mass function evaluated at value.
Attributes