MaartenGr / MaartenGr/BERTopic

How to improve Hierarchical Clustering

Open
#1,877 3 comments 0 reactions 0 assignees View on GitHub

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

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. 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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.