如何在tensorflow_word2vec中像gensim中使用model.most_similar一樣查找某個單詞的相似單詞？

Question

我使用tensorflow建立word2vec模型，參考這裏：https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/tutorials/word2vec/word2vec_basic.py#L118

我的問題是，我怎麼能找到前N個類似的話一定word.I知道gensim，我可以保存和加載word2vec模型，然後使用model.most_similar找到我想要的。但是如何在tensorflow中，甚至更多是否有任何方法來保存張量流模型，因爲我發現我得到的只是一個嵌入向量，是嗎？

Answer 1

我認爲只要你計算了每個標記的權重向量，那麼你可以操縱向量空間中的所有標記。您可以簡單地計算每個矢量之間的餘弦相似度，然後按照得分進行排序。爲了供您參考，您可以查看在gensim word2vec模型中實施的most_similar方法的source code。希望這可以幫助。