基于Bert的文本聚类工具:BERTopic

【干货推荐] 基于Bert的聚类工具:BERTopic
【简介】:BERTopic是一种主题建模技术,它利用变换器和c-TF-IDF创建聚类簇,使主题易于理解,同时在主题描述中保留重要的单词。 同时可以支持类似于LDAvis的可视化。
【快速上手】 安装pip install bertopic[visualization]

from bertopic import BERTopic
from sklearn.datasets import fetch_20newsgroups
 
docs = fetch_20newsgroups(subset='all',  remove=('headers', 'footers', 'quotes'))['data']

topic_model = BERTopic()
topics, _ = topic_model.fit_transform(docs)

【官网链接】:GitHub链接为https://github.com/MaartenGr/BERTopic,官方文档https://maartengr.github.io/BERTopic/index.html

基于Bert的文本聚类工具:BERTopic_第1张图片
基于Bert的文本聚类工具:BERTopic_第2张图片

你可能感兴趣的:(python,java,机器学习,github,linux)