【发布时间】:2017-07-22 02:41:22
【问题描述】:
我想使用 word2vec 的输出嵌入,例如 this paper (Improving document ranking with dual word embeddings)。
我知道输入向量在 syn0 中,如果负采样,输出向量在 syn1 和 syn1neg 中。
但是当我用输出向量计算 most_similar 时,由于删除了 syn1 或 syn1neg,我在某些范围内得到了相同的结果。
这是我得到的。
IN[1]: model = Word2Vec.load('test_model.model')
IN[2]: model.most_similar([model.syn1neg[0]])
OUT[2]: [('of', -0.04402521997690201),
('has', -0.16387106478214264),
('in', -0.16650712490081787),
('is', -0.18117375671863556),
('by', -0.2527652978897095),
('was', -0.254993200302124),
('from', -0.2659570872783661),
('the', -0.26878535747528076),
('on', -0.27521973848342896),
('his', -0.2930959463119507)]
但另一个 syn1neg numpy 向量已经是类似的输出。
IN[3]: model.most_similar([model.syn1neg[50]])
OUT[3]: [('of', -0.07884830236434937),
('has', -0.16942456364631653),
('the', -0.1771494299173355),
('his', -0.2043554037809372),
('is', -0.23265135288238525),
('in', -0.24725285172462463),
('by', -0.27772971987724304),
('was', -0.2979024648666382),
('time', -0.3547973036766052),
('he', -0.36455872654914856)]
我想在训练期间保留输出的 numpy 数组(无论是否为负)。
让我知道如何访问纯 syn1 或 syn1neg,或代码,或某些 word2vec 模块可以获得输出嵌入。
【问题讨论】:
标签: python numpy gensim word2vec