search

MaartenGr / BERTopic

Leveraging BERT and c-TF-IDF to create easily interpretable topics.
https://maartengr.github.io/BERTopic/
MIT License
6.18k stars 765 forks source link

FEATURE: Add interactive DataMapPlot #1853

Open dkapitan opened 8 months ago

dkapitan commented 8 months ago

https://github.com/MaartenGr/BERTopic/pull/1750/files# has integrated static datamapplot. The code base could be extended to support interactive datamapplots

Btibert3 commented 8 months ago

That's awesome.

MaartenGr commented 8 months ago

That definitely would be nice to implement. I'm wondering whether it could be done using the .visualize_document_datamap and simply add an interactive parameter of sorts or whether it deserves a separate function. @lmcinnes You by chance have any idea what would be best here?

lmcinnes commented 8 months ago

I think you can probably just add an interactive parameter, and have not too many other changes. It may be worth adding a few extra keyword parameters, specific to the interactive plot, that will be ignored if interactive is not True just to expose them in the API/docstring. I think enable_search would be the prime candidate, but there may be others.

dkapitan commented 8 months ago

thanks for the suggestion @lmcinnes. I will put this on my next item to pick up, @MaartenGr