Pytorch:使用nn.嵌入对分类特征进行编码

2024-04-26 00:02:38 发布

您现在位置:Python中文网/ 问答频道 /正文

我是个新手。我学习了一些关于如何使用pytorch构建简单NN模型的教程,例如(this one)。我发现他们更喜欢nn.嵌入来编码分类特征。我的问题是:(1)为什么不使用一个热编码?(2) 嵌入是随机初始化的,我认为嵌入的数值不会被训练,因为它们被当作输入。如果nn.Embedding随机初始化分类特征,会对训练产生负面影响吗


Tags: 模型编码分类教程nnpytorchembedding特征
1条回答
网友
1楼 · 发布于 2024-04-26 00:02:38

(1)预埋件通常尺寸较小且连续。这使他们更适合NN培训

(2)如果您刚刚初始化了它们,则应该对它们进行培训。它们被认为是模型的一部分,有点像连接一个热编码值与模型其余部分的连接点

相关问题 更多 >