MaartenGr / MaartenGr/BERTopic
How to improve Hierarchical Clustering
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 7.8k
- Forks
- 920
- Avg merge
- 22h 24m
- Merged PRs (30d)
- 5
Description
Hi,
After I trained my model, I wanted to make clusters on similar topic by using visualize_hierarchy but the result is not that great. The topics in each clusters are not really related. Is it because the dimension of the c_tf_idf matrix is so big ? Or is it because I used transformers to fine-tune my topic representation and so there is a high discrepancy between the topic keywords and topic representation that leads to bad Hierarchical Clustering ?
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with the reported visualize_hierarchy result and the c_tf_idf matrix and transformer-tuned topic representations described in the issue. Reproduce the clustering behavior, then determine whether the cause is the matrix dimensionality or the difference between topic keywords and representations. Done means identifying and documenting a clear cause or a concrete change needed to improve the hierarchy.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- data-visualization, machine-learning
- Issue type
- Bug
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100