如何从word2vec.py获得最终嵌入？

Question

我正在使用Google提供的word2vec.py，请参阅以下链接： https://github.com/tensorflow/tensorflow/blob/r0.7/tensorflow/models/embedding/word2vec.py

我想提取单词的最终嵌入并将其保存到txt文件中。我怎样才能找到最终的嵌入？

Answer 1

我想出了怎么做。

我在build_eval_graph模型中创建了一个self._nemb_final来传递nemb。 在训练模型后，我们可以做某事。如下所示：

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