Instructions to use MaartenGr/BERTopic_Wikipedia with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- BERTopic
How to use MaartenGr/BERTopic_Wikipedia with BERTopic:
from bertopic import BERTopic model = BERTopic.load("MaartenGr/BERTopic_Wikipedia") - Notebooks
- Google Colab
- Kaggle
What is the training benchmark for model `BERTopic_Wikipedia`
#3
by benjaminliupenrose - opened
Hello, I am trying to rerun the training for model with more customized training data. May I know for your training, what is
total training timeGPU spec and numbersCPU spec and numbersRAM
Thank you for giving us advice!
Thank you for your question! It was a while ago but I believe it took a couple of hours using a T4 GPU on a free google colab session. You can find a number of guides also here: https://github.com/MaartenGr/BERTopic?tab=readme-ov-file#getting-started
I believe I used the "Topic Modeling on Large Data" notebook.