Skip to content

Latest commit

 

History

History
35 lines (21 loc) · 1.17 KB

README.md

File metadata and controls

35 lines (21 loc) · 1.17 KB

Buzzwords

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

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 buzzwords

This will create the conda environment (with either your given name or buzzwords as default) with Buzzwords installed in it

Basic Examples

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)