Gensim: как загрузить предварительно обученную модель 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>()
----> 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'

python model gensim word2vec doc2vec
person fremorie    schedule 28.12.2016    source источник
comment
Я предполагаю, что это может быть проблема, связанная с форматом предварительно обученной / модели, вы можете использовать Doc2Vec, чтобы сохранить ее и загрузить, чтобы проверить, работает ли она.   -  person linpingta    schedule 28.12.2016
comment
Попробуйте обновить gensim из репозитория GitHub. Эта ошибка была исправлена ​​6 дней назад. github.com/RaRe-Technologies/gensim/issues/853   -  person Eduard Ilyasov    schedule 28.12.2016
comment
Кажется, это именно то, что мне нужно: модель была сохранена через Python 3, и я пытаюсь загрузить ее с помощью Python 2. Однако после того, как я обновил gensim из его репозитория GitHub, ничего не изменилось ...   -  person fremorie    schedule 28.12.2016
comment
Может я выбрал не ту ветку? Разве эта ошибка не исправлена ​​в ветке 'develop'?   -  person fremorie    schedule 28.12.2016
comment
Это исправление появилось в выпуске 0.13.4 и было объединено с основной веткой 3 дня назад. Вы можете найти свою проблему в списке изменений здесь, github.com/RaRe-Teimology/ совершить /. Вы можете попробовать удалить свой gensim и снова установить его из репозитория, следуя инструкциям из раздела «Установка».   -  person Eduard Ilyasov    schedule 28.12.2016
comment
где я могу найти сохраненную обученную модель?   -  person Chedi Bechikh    schedule 20.01.2017

Ответы (1)


arrow_upward
0
arrow_downward

Как отмечалось в комментариях к вопросу, это, вероятно, было связано с проблемой в gensim, которая была исправлена ​​в выпуске 0.13.4.

person gojomo    schedule 15.05.2017