Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add option for sequential prediction to `mcmc.predictive` #1995

neerajprad opened this issue Aug 6, 2019 · 0 comments


Copy link

commented Aug 6, 2019

Based on this forum post, I think a lot of new users of MCMC struggle with using plate handlers to correctly annotate their batch dimensions (which isn't strictly needed for doing inference).

While this isn't necessarily a requirement for MCMC, it becomes important if users would like to use the predictive utility to do predictions after running MCMC. This utility wraps the model in an outermost plate to vectorize predictions. I think we should support a parallel=False argument to this utility which by default will run the model forward on a per trace basis and only use plate if parallel is set to True. This will make it easier for new users to play around with MCMC who may not be well acquainted with our auto batching semantics.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
None yet
1 participant
You can’t perform that action at this time.