NumPyroを使ったzero-inflated distributionの実装
zero-inflated Poisson distributionとzero-inflated negative binomial distributionを使うためのNumPyroの実装例を作成しました。ご自由にご利用ください。いずれもGitHubのgistです。
Zero-inflated Poisson
https://gist.github.com/tomonari-masada/470541edbf9826ce3aec061c20df791c
Zero-inflated negative binomial
https://gist.github.com/tomonari-masada/e5ce480092acb275f6ff3f653d111c0d