Allow option to show heatmap annotation

[Iyeleon]

Feature request

Feature Request:
Add text_auto=True to BERTopic heatmap visualizations to automatically annotate cells with their values, similar to px.imshow. This improves readability without manual annotation.
Example: topic_model.visualize_heatmap(text_auto=True)

Motivation

• Improves readability of heatmaps without extra user effort.

Your contribution

.

[MaartenGr]

Do you perhaps have an example of how you would like to see the heatmaps changed? Perhaps a link to how px.imshow does this? This would help me understand what you are looking for.

[Iyeleon]

Hi Maarten, many thanks for your response. I have local code that updates the heatmap and other visualizations to work for me. They were pretty easy edits that I can share via a pull request in the coming week.

• I exposed text_auto = True in the argument for the heatmap, so i can pass it to px.imshow.
• I also exposed the colorscale argument so I can have control over the color.
• Finally rotated the text to 90 degrees and exposed an option to see the colorbar or not.

Having flexibility for heatmap and for other plotting functions makes it easier for me to integrate plot outputs into the application I'm working on.