Skip to content

Latest commit

 

History

History
20 lines (12 loc) · 386 Bytes

README.md

File metadata and controls

20 lines (12 loc) · 386 Bytes

installation

pip install git+https://github.com/the-deep-nlp/topic-generator

Usage

simple topic modeling utils based on BertTopic

from topic_generator import TopicGenerator

text = [YOUR TEXT EXCERPTS LIST]
embeddings = [YOUR CORRESPONDING VECTORS LIST]

topic_model = TopicGenerator(text, embeddings)
topic_model.get_total_topics()

topic_model.general_topics_df