scglue.models.prob.ZINB

class scglue.models.prob.ZINB(zi_logits, total_count, logits=None)

Bases: NegativeBinomial

Zero-inflated negative binomial distribution

Parameters:

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

• total_count (torch.Tensor) – Total count of the negative binomial distribution

• logits (typing.Optional[torch.Tensor]) – Logits of the negative binomial distribution

Methods

log_prob

Returns the log of the probability density/mass function evaluated at value.

Attributes