首页 PaddleNLP 帖子详情
如何保存词典和embedding table
收藏
快速回复
PaddleNLP 问答训练 2691 2
如何保存词典和embedding table
收藏
快速回复
PaddleNLP 问答训练 2691 2

在词向量训练完成后(例如http://www.paddlepaddle.org/documentation/docs/zh/1.2/beginners_guide/basics/word2vec/index.html),如何把词典和embedding table单独保存下来,后面可以直接使用?我在这里(http://aistudio.baidu.com/aistudio/#/projectdetail/180)找到v2版本的方法,但不知道fluid中如何实现。

0
收藏
回复
全部评论(2)
时间顺序
l
liuyibing0703
#2 回复于2019-02

可以参考这里 https://github.com/PaddlePaddle/models/blob/c1c5ed644428acc8395ea9ea28518a1c66d790b3/fluid/PaddleRec/word2vec/infer.py#L171

0
回复
A
AIStudio43700
#3 回复于2019-02
可以参考这里 https://github.com/PaddlePaddle/models/blob/c1c5ed644428acc8395ea9ea28518a1c66d790b3/fluid/PaddleRec/word2vec/infer.py#L171
展开

请问,得到的emb就是所有训练词的词向量的numpy矩阵吗?就相当于以前版本中embeddings=parameters.get("_proj").reshape(len(word_dict), embsize)中的embeddings吗?谢谢

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户