Skip to content

Add ZeroInflatedPoisson distribution - #2030

Open
mmustafasenoglu wants to merge 1 commit into
tensorflow:mainfrom
mmustafasenoglu:feat/zero-inflated-poisson
Open

Add ZeroInflatedPoisson distribution#2030
mmustafasenoglu wants to merge 1 commit into
tensorflow:mainfrom
mmustafasenoglu:feat/zero-inflated-poisson

Conversation

@mmustafasenoglu

Copy link
Copy Markdown

Fixes #1134

The ZeroInflatedNegativeBinomial distribution already exists via inflated_factory, but the Zero-Inflated Poisson variant was missing. This adds ZeroInflatedPoisson using the same factory pattern, completing the feature request in #1134.

Changes:

  • inflated.py: Added ZeroInflatedPoisson via inflated_factory
  • __init__.py: Export ZeroInflatedPoisson
  • inflated_test.py: Added tests for basic sampling, log_prob, logits mode, and batched inputs

Fixes tensorflow#1134

The ZeroInflatedNegativeBinomial distribution already exists via
inflated_factory, but the Zero-Inflated Poisson variant was missing.
This adds ZeroInflatedPoisson using the same factory pattern,
completing the feature request in tensorflow#1134.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[Feature Request] Zero-Inflated Poisson and Negative Binomial distributions

1 participant