pyLDAvis visualization from gensim not displaying the result in google colab

Question

import pyLDAvis.gensim
# Visualize the topics
pyLDAvis.enable_notebook()
vis = pyLDAvis.gensim.prepare(lda_model, corpus, id2word)
vis

The above code displayed the visualization of LDA model in google colab but then after reopening the notebook it stopped displaying. I even tried pyLDAvis.display(vis, template_type='notebook') still not working

When I set

pyLDAvis.enable_notebook(local=True)

it does display the result but not the labels.. Any help would be appreciated!!

Answer 1

when you install LDAvis make sure to specify the version to be 2.1.2 with:

!pip install pyLDAvis==2.1.2

the new versions don't seem to play well with colab.

Answer 2

they changed the package name. use it like:

import pyLDAvis.gensim_models
vis = pyLDAvis.gensim_models.prepare(lda_model, corpus, id2word)
vis