无法加载保存的gensim word2vec mod

2024-09-28 21:45:45 发布

您现在位置:Python中文网/ 问答频道 /正文

我试着保存我用gensim训练过的word2vec模型,如下所示:

from gensim.models import Word2Vec
model = Word2Vec(sentences, parameters)
model.save('modelfile.model')

现在,当我尝试Word2Vec.load('modelfile.model')时,我得到:

^{pr2}$

如果有帮助,可以发布完整的回溯。在


Tags: from模型importmodelmodelssavesentencesload
2条回答

我会尝试virtualenv来避免包和环境之间的任何冲突。Virtualenv是一个创建独立Python环境的工具。下面是设置virtualenv并尝试加载word2vec模型的快速步骤。在

sudo pip install virtualenv # install virtualenv
virtualenv test1 # create an env named test1
cd test1 # go ot test1
source bin/activate # activate test1
pip install gensim # install your packages for test1 env
python path-to-your-file.py # run your program 

真奇怪。您对load()使用与save()完全相同的Python环境和gensim版本吗?你是如何安装gensim&numpy的?在

当我搜索这个错误时,我发现其他讨论表明这可能是从numpy 1.16中pickle(保存)了一个numpy数组的症状。但在之前的一次裸体游戏中试图解开它。示例见:

https://github.com/numpy/numpy/issues/12825#issuecomment-456561919

看起来numpy最近合并了一个修复程序&{a2},但与此同时,最好的办法可能是确保加载的位置使用的是numpy 1.16.0+。在

相关问题 更多 >