Calculate log probability under zero-inflated negative binomial.
Adapted from scvi-tools.
Usage
log_zinb(x, mu, theta, pi, eps = 1e-08)
Arguments
- x
Observed counts
- mu
Mean parameter
- theta
Dispersion parameter
- pi
Dropout probability (logit scale)
- eps
Small constant for numerical stability
Value
Tensor of log probabilities
References
Gayoso et al. scvi-tools. https://github.com/YosefLab/scvi-tools