Получить весовые матрицы от gensim word2Vec

Question

Я использую пакет gensim word2vec в python. Я хотел бы получить весовые матрицы W и W', которые были изучены во время обучения методам пропуска.

Мне кажется, что model.syn0 дает мне первый, но я не уверен, как я могу получить другой. Есть идеи?

Я бы на самом деле хотелось найти исчерпывающую документацию по моделям доступных атрибутов, так как официальная, кажется, не быть точным (например, syn0 не описывается как атрибут)

Answer 1

model.wv.syn0 содержит ввода вложение матрица. Выход вложение хранится в model.syn1, когда он обучен hierarchical softmax (hs=1) или в model.syn1neg, когда он использует отрицательный отбор проб (negative>0). Это оно! Когда и иерархический softmax и отрицательный выборки не включены, Word2Vec использует единую весовую матрицу model.wv.syn0 для обучения.

См. Также связанную дискуссию here.