Tensorflow 如何从word2vec.py获取最终嵌入?

Tensorflow 如何从word2vec.py获取最终嵌入?,tensorflow,word2vec,Tensorflow,Word2vec,我正在使用谷歌的word2vec.py,请参见以下链接: 我想提取单词的最终嵌入并将其保存到txt文件中。如何找到最终嵌入?我找到了方法 我创建了一个self._nemb_final in build_eval_图形模型,将nemb转出。 模型训练完成后,我们可以做如下事情: final_embedding = self._nemb_final.eval() np.savetxt('/.../final_embedding_dic.txt',final_embedding) @ta

我正在使用谷歌的word2vec.py,请参见以下链接:


我想提取单词的最终嵌入并将其保存到txt文件中。如何找到最终嵌入?

我找到了方法

我创建了一个self._nemb_final in build_eval_图形模型,将nemb转出。 模型训练完成后,我们可以做如下事情:

final_embedding = self._nemb_final.eval()     
np.savetxt('/.../final_embedding_dic.txt',final_embedding)

@tanjir word2vec是在一些嵌入空间中用向量表示单词。嵌入向量通过一些神经网络进行学习/更新。在word2vec.py中,他们通过一些类比问题来测试结果。现在,我想提取语料库中所有单词的最终嵌入向量。这样就可以保存向量了。词汇表呢?@Michelle Owen:如何使用vocabulory获取向量?@ThomasNT您只需将词汇表与嵌入向量连接在一起即可。@开拓者请参见上文