Pobierz macierze wagowe z gensim word2Vec

Question

Używam pakietu word2vec gensim w python. Chciałbym odzyskać macierze wagowe W i W', które zostały nauczone podczas nauki gry na wątku.

Wydaje mi się, że model.syn0 daje mi pierwszą, ale nie jestem pewien, jak mogę uzyskać drugą. Dowolny pomysł?

I rzeczywiście kochać znaleźć żadnej wyczerpującej dokumentacji dotyczącej modeli dostępnych atrybutów, ponieważ oficjalna jeden nie wydaje się być precyzyjny (na przykład syn0 nie jest opisany jako atrybut)

Answer 1

model.wv.syn0 zawiera wejściowego osadzanie matryca. Osadzanie danych wyjściowych jest przechowywane w po przeszkoleniu pod numerem hierarchical softmax (hs=1) lub w model.syn1neg, gdy używa ujemnego próbkowania (negative>0). To jest to! Kiedy zarówno hierarchiczna miękka, jak i ujemna próbka nie są włączone, Word2Vec używa pojedynczej macierzy wagi model.wv.syn0 do treningu.

Zobacz także pokrewną dyskusję here.