Gensim : pretrained doc2vec 모델을로드하는 방법?

내 doc2vec pretrained 모델을 읽으려고 해요 :Gensim : pretrained 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'

출처

2016-12-28 fremorie

"pretrained/model"형식과 관련된 문제가있을 수 있습니다. Doc2Vec을 사용하여 파일을 저장하고 작동하는지 테스트하십시오. – linpingta

GitHub repo에서 gensim을 업데이트하십시오. 이 버그는 6 일 전에 고쳐졌습니다. https://github.com/RaRe-Technologies/gensim/issues/853 –

정확하게 필요한 것 같습니다 : 모델이 파이썬 3을 통해 저장된 것처럼 보였습니다. 파이썬 2로로드하려고합니다. 그러나, GitHub 레포에서 gensim을 업데이트 한 후에는 아무 것도 변경되지 않았습니다 ... – fremorie

이 질문에 대한 의견에서 언급했듯이 이것은 0.13.4 릴리스에서 수정 된 gensim의 문제와 관련이있을 수 있습니다.

출처

2017-05-15 23:02:46 gojomo

Gensim : pretrained doc2vec 모델을로드하는 방법?

답변

관련 문제