probml / pyprobml

Python code for "Probabilistic Machine learning" book by Kevin Murphy
MIT License
6.45k stars 1.52k forks source link

fix VBEM for GMM (Gerardo's JAX code) #1072

Closed murphyk closed 2 years ago

murphyk commented 2 years ago

The demo https://github.com/probml/pyprobml/blob/master/notebooks/book2/10/variational_mixture_gaussians_demo.ipynb does not run because cannot import name 'index_update'. (The notebook uses https://github.com/probml/probml-utils/blob/main/probml_utils/variational_mixture_gaussians.py)