MaartenGr/BERTopic

Custom labels for get_topic_tree

Open

#2123 aperta il 17 ago 2024

Vedi su GitHub
 (2 commenti) (0 reazioni) (0 assegnatari)Python (634 fork)batch import
good first issue

Metriche repository

Star
 (5074 star)
Metriche merge PR
 (Nessuna PR mergiata in 30 g)

Descrizione

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!

Guida contributor

Direzione di ricerca
Aggiungere un parametro `custom labels` al metodo `get topic tree`, seguendo l'implementazione in `visualize hierarchy`. Ciò comporta la modifica della firma del metodo per accettare una lista opzionale di etichette e utilizzare quelle etichette al posto delle rappresentazioni predefinite degli argomenti durante la costruzione della struttura ad albero.
Tech stack
python
Dominio
machine learning
Tipo issue
Funzionalità
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
2
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-3 ore
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Abbastanza chiara
Prerequisiti
PythonBERTopic
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
90