我想自己训练标签嵌入,(是的,标签嵌入像字嵌入,但输入是标签的一个热矢量)
当我发现this
我在官方文档中找到了示例时,它必须通过W。
如何训练嵌入W矩阵,那么以后我们可以用它来训练另一个模型?
我的意思是,如何训练嵌入字/标签的矢量表示?
答案 0 :(得分:0)
您不需要采取两步(火车嵌入,然后训练另一个模型),但您可以以端对端的方式训练嵌入。 一旦你从分类值中嵌入了矢量嵌入,就可以将它连接到通常的神经网络来训练损失。
Word2vec是使用EmbedID
的一个官方示例: