Python Gensim:如何加载预训练的doc2vec模型?

Python Gensim:如何加载预训练的doc2vec模型?,python,model,gensim,word2vec,doc2vec,Python,Model,Gensim,Word2vec,Doc2vec,我正在阅读我的预训练doc2vec模型: from gensim.models import Doc2Vec model = Doc2Vec.load('/path/to/pretrained/model') 但是,在读取过程中会出现错误。有人能建议如何处理这个问题吗?以下是错误: AttributeErrorTraceback (most recent call last) <ipython-input-9-819b254ac835> in <module>() --

我正在阅读我的预训练doc2vec模型:

from gensim.models import Doc2Vec
model = Doc2Vec.load('/path/to/pretrained/model')
但是,在读取过程中会出现错误。有人能建议如何处理这个问题吗?以下是错误:

AttributeErrorTraceback (most recent call last)
<ipython-input-9-819b254ac835> in <module>()
----> 1 model = Doc2Vec.load('/path/to/pretrained/model')

/opt/jupyter-notebook/.local/lib/python2.7/site-packages/gensim/models/word2vec.pyc in load(cls, *args, **kwargs)
   1682     @classmethod    
   1683     def load(cls, *args, **kwargs):
-> 1684         model = super(Word2Vec, cls).load(*args, **kwargs)
   1685         # update older models
   1686         if hasattr(model, 'table'):

/opt/jupyter-notebook/.local/lib/python2.7/site-packages/gensim/utils.pyc in load(cls, fname, mmap)
    246         compress, subname = SaveLoad._adapt_by_suffix(fname)
    247 
--> 248         obj = unpickle(fname)
    249         obj._load_specials(fname, mmap, compress, subname)
    250         return obj

/opt/jupyter-notebook/.local/lib/python2.7/site-packages/gensim/utils.pyc in unpickle(fname)
    909     with smart_open(fname) as f:
    910         # Because of loading from S3 load can't be used (missing readline in smart_open)
--> 911         return _pickle.loads(f.read())
    912 
    913 

AttributeError: 'module' object has no attribute 'defaultdict'
AttributeErrorTraceback(最近一次调用上次)
在()
---->1 model=Doc2Vec.load(“/path/to/pretrained/model”)
/opt/jupyter notebook/.local/lib/python2.7/site-packages/gensim/models/word2vec.pyc加载(cls,*args,**kwargs)
1682@classmethod
1683 def加载(cls、*ARG、**kwargs):
->1684型号=超级(Word2Vec,cls)。加载(*args,**kwargs)
1685#更新旧型号
1686如果hasattr(型号“表”):
/opt/jupyter notebook/.local/lib/python2.7/site-packages/gensim/utils.pyc加载(cls、fname、mmap)
246压缩,子名称=保存加载。_通过_后缀(fname)进行调整
247
-->248 obj=取消勾选(fname)
249对象加载特殊项(fname、mmap、compress、subname)
250返回obj
/unpickle(fname)中的opt/jupyter notebook/.local/lib/python2.7/site-packages/gensim/utils.pyc
909,智能开启(fname)为f:
910#由于从S3加载,无法使用加载(smart#u open中缺少读线)
-->911 return_pickle.load(f.read())
912
913
AttributeError:“模块”对象没有属性“defaultdict”

如问题评论中所述,这可能与gensim中的一个问题有关,该问题在0.13.4版本中已修复

如问题评论中所述,这可能与gensim中的一个问题有关,该问题在0.13.4版本中已修复

我想这可能是与“pretrained/model”的格式有关的问题,您可以使用Doc2Vec保存一个,然后加载以测试它是否可以从GitHub repo更新gensim。此错误已在6天前修复。这似乎正是我所需要的:模型似乎是通过Python3保存的,我正在尝试用Python2加载它。然而,在我从GitHub repo更新gensim之后,一切都没有改变……也许我选择了错误的分支?这个bug不是在“开发”分支中修复的吗?这个修复出现在0.13.4版本中,并在3天前与主分支合并。您可以在此处的更改列表中找到您的问题。您可以尝试删除您的gensim,然后按照“安装”中的说明再次从repo中安装它。我猜可能是与“pretrained/model”的格式有关的问题,您可以使用Doc2Vec保存一个并加载以测试它是否工作。尝试从GitHub repo更新gensim。此错误已在6天前修复。这似乎正是我所需要的:模型似乎是通过Python3保存的,我正在尝试用Python2加载它。然而,在我从GitHub repo更新gensim之后,一切都没有改变……也许我选择了错误的分支?这个bug不是在“开发”分支中修复的吗?这个修复出现在0.13.4版本中,并在3天前与主分支合并。您可以在此处的更改列表中找到您的问题。您可以尝试删除gensim,然后按照“安装”中的说明再次从repo安装它。