【干货推荐] 基于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
网友评论