标签:

• bertopic 库名称: bertopic 流水线标签: 文本分类 许可证: mit 数据集:
• OpenAssistant/oasst1 语言:
• en

聊天话题模型

这是一个基于BERTopic的模型。
BERTopic是一个灵活模块化的话题建模框架，能够从海量数据中生成易于解释的话题分类。

使用方法

安装BERTopic库：

pip install -U bertopic

调用模型示例：

from bertopic import BERTopic
topic_model = BERTopic.load("davanstrien/chat_topics")

topic_model.get_topic_info()

话题概览

• 话题总数：75个
• 训练文档量：63530篇

训练超参数

• 概率计算：关闭
• 语言：未指定
• 低内存模式：关闭
• 最小话题规模：20
• N元语法范围：(1,1)
• 话题数量：75
• 种子话题列表：无
• 关键词数量：10
• 详细日志：开启

框架版本

• Numpy：1.22.4
• HDBSCAN：0.8.29
• UMAP：0.5.3
• Pandas：1.5.3
• Scikit-Learn：1.2.2
• Sentence-transformers：2.2.2
• Transformers：4.29.2
• Numba：0.56.4
• Plotly：5.13.1
• Python：3.10.11