BERTopic 嵌入错误

tyky79it  于 6个月前  发布在  其他
关注(0)|答案(3)|浏览(81)

你好,@MaartenGr,
我安装了一个谷歌包,更新了一些软件包,之后出现了以下错误。你能帮我解决这个问题吗?谢谢!
2023-09-19 16:43:58,838 - BERTopic - 将文档转换为嵌入向量
回溯(最近的调用):
topics, probs = topic_model.fit_transform(docs)
文件 ".../bertopic/_bertopic.py",第350行,在 fit_transform 中
y, embeddings = self._guided_topic_modeling(embeddings)
文件 ".../bertopic/_bertopic.py",第2919行,在 _guided_topic_modeling 中
seed_topic_embeddings = np.vstack([seed_topic_embeddings, embeddings.mean(axis=0)])
文件 "<array_function internals>",第5行,在 vstack 中
文件 ".../site-packages/numpy/core/shape_base.py",第282行,在 vstack 中
返回 _nx.concatenate(arrs, 0)
文件 "<array_function internals>",第5行,在 concatenate 中
ValueError: 所有输入数组沿连接轴的维度必须完全匹配,但在维度1上,索引0处的数组大小为46,索引1处的数组大小为100

hkmswyz6

hkmswyz61#

可能是numpy和/或numba更新到了不兼容的版本。回到之前的环境看看内部发生了什么变化可能会有所帮助。

zaqlnxep

zaqlnxep2#

我遇到了类似的错误。请问您测试了哪个版本的numpy、scipy和numba?谢谢!

zbdgwd5y

zbdgwd5y3#

你好@cosmozhang
我正在使用以下版本的numpy和numba来恢复BERTopic。
numpy=1.21.1
numba=0.53.1

相关问题