Fatal编程技术网
  • tensorflow/
  • 如何在word2vec.py（tensorflow）中检索每个嵌入的对应单词？

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

tensorflow

如何在word2vec.py（tensorflow）中检索每个嵌入的对应单词？,tensorflow,word2vec,Tensorflow,Word2vec,py所做的是返回一个形状张量（vocab_大小、嵌入_大小）。我想要每个嵌入的对应单词，有点像字典： { word1 : embedding1 , word2 : embedding2 .... } 如何获取它？您可以使用word2vec.\u id2word获取id的索引。您可以使用word2vec.\u id2word获取id的索引

py所做的是返回一个形状张量（vocab_大小、嵌入_大小）。我想要每个嵌入的对应单词，有点像字典：

{ word1 : embedding1 , word2 : embedding2 .... }
如何获取它？

您可以使用word2vec.\u id2word获取id的索引。

您可以使用word2vec.\u id2word获取id的索引