Skip to main content

Module vb

Module vb 

Source
Expand description

The variational Bayes solver: E-step, M-step, bound and the outer loop.

Both steps are fan-outs over disjoint output slices, so neither needs a reduction. The E-step gives each document its own column of gamma; the M-step gives each term its own column of the sufficient statistics.

§References

Hoffman, Blei and Bach, Online Learning for Latent Dirichlet Allocation, NIPS, 2010

Functions§

lda_bound
Evaluate the variational bound for an externally supplied set of parameters.