sccross.models.utils.ZINB

class sccross.models.utils.ZINB(zi_logits, total_count, logits=None)[source]

Bases: torch.distributions.negative_binomial.NegativeBinomial

Zero-inflated negative binomial distribution

Parameters

  • zi_logits (Tensor) – Zero-inflation logits

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

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

Methods

log_prob

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

Attributes