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

BERTopic初始化报错:__init__()不支持'diversity'参数求助

问题解决:BERTopic初始化报diversity参数错误

问题原因

当前使用的BERTopic版本中,diversity不再是BERTopic类初始化的参数——该参数在版本迭代中被移到了reduce_topics方法里,而你的代码还在初始化时传入这个参数,因此触发TypeError。

解决方案

方案1:回退到支持初始化传diversity的旧版本

运行以下命令安装指定旧版本(以0.14.1为例,该版本支持初始化时传入diversity):

pip install bertopic==0.14.1

方案2:修改代码适配新版本

  1. 先修改BERTopic初始化代码,移除diversity=diversity参数:
topic_model = BERTopic(min_topic_size=min_topic_size, #Modificar si hay errores
                       top_n_words=top_n_words, 
                       verbose=True, #Informe de resultados
                       #language="multilingual", 
                       calculate_probabilities = True, #desactivar si hay muchos datos
                       vectorizer_model = vectorizer_model,
                       embedding_model = sentence_model,
                       low_memory = low_memory
                      ) #Añadir low_memory=True si hay muchos datos 
  1. 在模型训练完成(调用fit_transform后),添加主题多样性调整的代码:
# 假设documents是你的训练文本数据
topics, probabilities = topic_model.fit_transform(documents)
# 传入diversity参数调整主题多样性
topic_model = topic_model.reduce_topics(documents, diversity=diversity)

内容的提问来源于stack exchange,提问作者Junior Estrella

相关产品推荐
方舟 Agent Plan

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

最近更新时间:2026.07.29 09:25:41