MaartenGr / MaartenGr/BERTopic

Setting up an end-to-end system with BERTopic efficiently.

Open
#1,475 1 comment 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 Maarten,

Thanks for developing such a modular topic modeling setup and the awesome library!
I have been trying to setup a system with BERTopic which performs the following steps:

1. Given a set of starting documents, train a BERTopic model to get relevant topics and create a `{topic_id: topic}` map
2. Given new documents, perform inference on them to assign them to existing _topic_ids_
3. Train with only the new documents to assign them to previous _topic_ids_ and create new _topic_ids_

My questions are as follows:

- I have some documents that are larger than the length of what `sentence transformers` accept and would like to chunk them. I am not sure how to maintain an `id` for the document with BERTopic such that after training/inference I can associate multiple topics and thus _topic_ids_ with the document.
- Do you have recommendations for training the model in step 3 such that we do not need to re-train with documents used in _step_1_?
- Any other suggestions you might have for the steps outlined above!

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 by reviewing the BERTopic workflow described in the issue, including document chunking, inference against existing topic IDs, and incremental training. The issue is complete when the project provides documented guidance for preserving document associations and updating a model without retraining on the original documents.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Documentation
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
20/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.