scglue.models.prob.ZINB
- class scglue.models.prob.ZINB(zi_logits, total_count, logits=None)[source]
Bases:
NegativeBinomial
Zero-inflated negative binomial distribution
- Parameters:
zi_logits (
torch.Tensor
) – Zero-inflation logitstotal_count (
torch.Tensor
) – Total count of the negative binomial distributionlogits (
typing.Optional
[torch.Tensor
]) – Logits of the negative binomial distribution
Methods
Returns the log of the probability density/mass function evaluated at value.
Attributes