MaartenGr / BERTopic

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

Add custom labels to get topic tree #2125

Open clfkenny opened 3 months ago

clfkenny commented 3 months ago

What does this PR do?

get_topic_tree function currently does not accept custom_labels as a parameter whereas the other visualization methods do. Therefore this change aims to fix that issue so custom labels can be accepted and make the visualization more intuitive.

I really enjoyed looking through the codebase here. My first PR here so still learning the project structure, would appreciate any feedback. Thanks!

Fixes # https://github.com/MaartenGr/BERTopic/issues/2123

Before submitting