MaartenGr/BERTopic

Custom labels for get_topic_tree

Open

#2.123 geöffnet am 17. Aug. 2024

Auf GitHub ansehen
 (2 Kommentare) (0 Reaktionen) (0 zugewiesene Personen)Python (634 Forks)batch import
good first issue

Repository-Metriken

Stars
 (5.074 Stars)
PR-Merge-Metriken
 (Keine gemergten PRs in 30 T)

Beschreibung

Discussed in https://github.com/MaartenGr/BERTopic/discussions/2122

Originally posted by clfkenny August 17, 2024 Hi there, I noticed the visualize_hierarchy method has a custom_labels parameter. Is there anything similar to this for the get_topic_tree method? I use the LLM generated labels as the custom labels so the hierarchy tree is much easier to interpret. Thanks!

This seems like a good first issue for someone wanting to contribute to this repository. It has a very similar structure to the visualize_hierarchy functionality, so I believe it should be straightforward to implement. Unfortunately, I have a large backlog and cannot prioritize this, so if anyone wants to help out then that would be great!

Contributor Guide