MaartenGr/BERTopic

Custom labels for get_topic_tree

Open

#2 123 ouverte le 17 août 2024

Voir sur GitHub
 (2 commentaires) (0 réactions) (0 assignés)Python (634 forks)batch import
good first issue

Métriques du dépôt

Stars
 (5 074 stars)
Métriques de merge PR
 (Aucune PR mergée en 30 j)

Description

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!

Guide contributeur

Direction de recherche
Ajoutez un paramètre `custom labels` à la méthode `get topic tree`, en reproduisant l'implémentation dans `visualize hierarchy`. Cela implique de modifier la signature de la méthode pour accepter une liste facultative d'étiquettes et d'utiliser ces étiquettes à la place des représentations de sujets par défaut lors de la construction de la structure arborescente.
Stack technique
python
Domaine
machine learning
Type d'issue
Fonctionnalité
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
2
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-3 heures
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Plutôt claire
Prérequis
PythonBERTopic
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
90