如何在chainer中训练chainer.links.EmbedID?

时间:2018-03-10 09:14:13

标签: chainer

我想自己训练标签嵌入,(是的,标签嵌入像字嵌入,但输入是标签的一个热矢量)

当我发现this我在官方文档中找到了示例时,它必须通过W。

如何训练嵌入W矩阵,那么以后我们可以用它来训练另一个模型?

我的意思是,如何训练嵌入字/标签的矢量表示?

1 个答案:

答案 0 :(得分:0)

您不需要采取两步(火车嵌入,然后训练另一个模型),但您可以以端对端的方式训练嵌入。 一旦你从分类值中嵌入了矢量嵌入,就可以将它连接到通常的神经网络来训练损失。

Word2vec是使用EmbedID的一个官方示例: