acerbilab / pyvbmc

PyVBMC: Variational Bayesian Monte Carlo algorithm for posterior and model inference in Python
https://acerbilab.github.io/pyvbmc/
BSD 3-Clause "New" or "Revised" License
114 stars 6 forks source link

feat: plot vp and vbmc iterations #32

Closed Solosneros closed 2 years ago

Solosneros commented 2 years ago

With this PR I add the possibility to both plot a vp with vp.plot() (the vp.plot() can be styled. See the documentation for that.) and also to plot a whole vbmc iteration run with the new function create_vbmc_animation(). (See the gif below).

Other changes include:

Note that the creation of the gif requires the package imageio, doing the same with matploblib was not possible for me. @lacerbi Let me know if that is a problem. Maybe we can figure out a better way. Also not sure how to test that function.

rosenbrock