BERTopic初始化报错:__init__()不支持'diversity'参数求助
问题解决:BERTopic初始化报
diversity参数错误 问题原因
当前使用的BERTopic版本中,diversity不再是BERTopic类初始化的参数——该参数在版本迭代中被移到了reduce_topics方法里,而你的代码还在初始化时传入这个参数,因此触发TypeError。
解决方案
方案1:回退到支持初始化传diversity的旧版本
运行以下命令安装指定旧版本(以0.14.1为例,该版本支持初始化时传入diversity):
pip install bertopic==0.14.1
方案2:修改代码适配新版本
- 先修改
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
- 在模型训练完成(调用
fit_transform后),添加主题多样性调整的代码:
# 假设documents是你的训练文本数据 topics, probabilities = topic_model.fit_transform(documents) # 传入diversity参数调整主题多样性 topic_model = topic_model.reduce_topics(documents, diversity=diversity)
内容的提问来源于stack exchange,提问作者Junior Estrella
相关产品推荐
相关产品推荐

