Buzzwords is Bumble's GPU-powered topic modelling tool, used for gathering insights on topics in text or images on a large scale. The algorithm is based on Bertopic and Top2Vec, but altered to be faster.
For more information see the website
Installation for buzzwords is somewhat complicated, due to the need for RAPIDS.ai (and to a lesser extent, FAISS) on an Nvidia GPU-powered machine. RAPIDS doesn't support installation through pip anymore, so we need to use conda environments.
For ease of installation, we've packaged it up into a bash script install.sh
$ ./install.sh buzzwordsThis will create the conda environment (with either your given name or buzzwords as default) with Buzzwords installed in it
To instantiate the model is very simple
from buzzwords import Buzzwords
model = Buzzwords()To train the model on a set of documents, call the fit_transform() function to return the topics
docs = df['text_column']
topics = model.fit_transform(docs)