如何在word2vec.py（tensorflow）中检索每个嵌入的相应单词？

Question

word2vec.py的作用是返回一个形状的张量（vocab_size，embedding_size）。我想要每个嵌入的相应单词，sorta就像字典：

{ word1 : embedding1 , word2 : embedding2 .... }

我如何得到它？

Answer 1

您可以使用word2vec._id2word获取id的索引。