我在单独的文件夹中有10个类的训练图像。(从0到9)我对标记数据感到震惊。我是否需要为每个图像或每个类遵循一个热编码?或任何……
你提到过你 10 类。您可以用任何名称标记它们。例如 class 1 : 'A', class '2' : 'B' 等等。
10
class 1 : 'A', class '2' : 'B'
但是在训练深度学习模型时,您必须对标签进行编码。正如您所提到的,您熟悉二进制分类,其中输入标签也是 0 要么 1 对于 cats 和 dogs 分类,这里的标签必须是 0 , 1 , 2 ,....., 9 。
0
1
cats
dogs
2
9
要训练模型,您不能直接使用此标签。您必须对这些标签进行单热编码并将其提供为 y 为您的模型。你的模特必须有 10 输出层中的节点。 argmax() 关于模型做出的预测将告诉您测试数据属于哪个类。
y
argmax()