gpt4 book ai didi

tensorflow - 我怎样才能从 word2vec.py 得到最终的嵌入?

转载 作者:行者123 更新时间:2023-12-03 11:23:33 25 4
gpt4 key购买 nike

我使用的是 Google 的 word2vec.py,请参阅以下链接: https://github.com/tensorflow/tensorflow/blob/r0.7/tensorflow/models/embedding/word2vec.py

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

最佳答案

我知道怎么做了。

我在 build_eval_graph 模型中创建了一个 self._nemb_final 来将 nemb 传出。模型训练好后,我们就可以做某事了。如下所示:

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

关于tensorflow - 我怎样才能从 word2vec.py 得到最终的嵌入?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/35582858/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com