python - 如何处理pytorch train、test和holdout set中的不同类别

Question

我有一个表格 pytorch 模型，它将城市和邮政编码作为分类嵌入。但是，我无法根据这些列进行有效分层。

如果 pytorch 在测试集中缺少不在训练集中的分类值，或者在保持集中没有在训练/测试集中的分类值，我如何让 pytorch 运行？

标签： pythonpytorch

Answer 1

你可以尝试使用一种热编码

PS：这是一个建议不是答案