加载BERTopic模型时出现问题,'NNDescent'对象没有'_bit_trees'属性

j2qf4p5b 于 3个月前发布在其他

关注(0)|答案(3)|浏览(66)

代码：

from bertopic import BERTopic
model_path = "/path/to/bertopic/model"
model = BERTopic.load(model_path)

错误：
AttributeError: 'NNDescent' object has no attribute '_bit_trees'
解决方案：
我将我的pynndescent从0.5.12降级到0.5.11。这解决了问题，但只是提醒大家注意，以防将来需要做出更改。

3条答案

感谢您的分享！您可能还可以通过安装最新版本的umap-learn来解决此问题，该版本可能已修复了此问题。

我现在也遇到了同样的问题，降级pynndescent或者安装umap-learn都没有帮助。之前我也遇到了加载模型的问题，为此我升级了numba。有什么建议吗？提前谢谢！

@lucijakrusic 不太确定这里的解决方法是什么。我假设你是从一个 pickle 加载的，对吗？如果是这样的话，你需要确保所有依赖项的所有版本都与保存模型时完全相同。