2016-12-28 6 views
1

Ich versuche, meine vortrainierter doc2vec Modell zu lesen:Gensim: Wie man vortrained doc2vec Modell lädt?

from gensim.models import Doc2Vec 
model = Doc2Vec.load('/path/to/pretrained/model') 

scheint jedoch ein Fehler während des Prozesses zu lesen. Könnte jemand vorschlagen, wie man damit umgeht? Hier ist der Fehler:

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' 
+0

Ich denke, es kann ein Problem mit dem Format "vortrainiert/Modell" sein, können Sie Doc2Vec verwenden, um eine zu speichern und laden, um zu testen, ob es funktioniert – linpingta

+1

Versuchen Sie, Gensim von GitHub Repo zu aktualisieren. Dieser Fehler wurde vor 6 Tagen behoben. https://github.com/RaRe-Technologies/gensim/issues/853 –

+0

Scheint genau das zu sein, was ich brauche: Das Modell scheint über Python 3 gesichert zu sein, und ich versuche, es mit Python 2 zu laden. nachdem ich Gensim von seinem GitHub Repo aktualisiert habe, hat sich nichts geändert ... – fremorie

Antwort

0

Wie in den Kommentaren zu der Frage erwähnt, war dies wahrscheinlich auf ein Problem in Gensim, das in Version 0.13.4 behoben wurde.

Verwandte Themen