You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

Biterm主题建模:如何获取原始格式的主题Top文档并导出?

解决方案:导出tmplot主题对应原始评论

直接可用代码(适配Python+BERTopic场景)

tmplot通常与BERTopic配合使用,以下代码假设你已完成主题建模,拥有topic_model建模对象和原始评论数据集docs:

import pandas as pd

# 获取各主题的Top原始文档索引及匹配分数
top_docs_info = topic_model.get_topics()

# 整理原始评论数据
export_records = []
for topic_id, doc_details in top_docs_info.items():
    for score, doc_index in doc_details:
        export_records.append({
            "主题ID": topic_id,
            "匹配分数": round(score, 4),
            "原始评论": docs[doc_index]
        })

# 导出为CSV文件(支持中文)
pd.DataFrame(export_records).to_csv("主题原始评论对照表.csv", index=False, encoding="utf-8-sig")

代码无法运行时的学习路径

  • 先掌握Python基础语法,重点熟悉pandas库的读写数据操作
  • 学习BERTopic的基础建模流程,理解topic_model对象的核心方法(如get_topics())
  • 查阅tmplot官方文档,明确主题与原始文档的关联逻辑

内容的提问来源于stack exchange,提问作者Alexander Avila

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.08.10 11:35:38