KitchenTable99/topic-wizard

Powerful topic model visualization in Python

topicwizard

Pretty and opinionated topic model visualization in Python.

topicwizard-2023-02-19_19.25.33.mp4

Features

• Investigate complex relations between topics, words and documents
• Highly interactive
• Name topics
• Pretty 🎨
• Intuitive 🐮
• Clean API 🍬
• Sklearn compatible 🔩
• Easy deployment 🌍

Installation

Install from PyPI:

pip install topic-wizard

Usage (documentation)

Step 1:

Train a scikit-learn compatible topic model.

from sklearn.decomposition import NMF
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.pipeline import Pipeline

topic_pipeline = Pipeline(
    [
        ("bow", CountVectorizer()),
        ("nmf", NMF(n_components=10)),
    ]
)
topic_pipeline.fit(texts)

Step 2:

Visualize with topicwizard.

import topicwizard

topicwizard.visualize(pipeline=topic_pipeline, corpus=texts)

Step 3:

Investigate 👀 .

a) Topics

b) Words

c) Documents