tensorflow - 使用稀疏_分类_交叉熵时如何定义自定义标签值？

Question

我的模型是用这段代码编译的

model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['sparse_categorical_accuracy'])

在训练期间，我收到此错误

tensorflow.python.framework.errors_impl.InvalidArgumentError: Received a label value of 5 which is outside the valid range of [0, 5).

我的标签是1,2,3,4,5，它是[1,5]而不是[0, 5)。如何为此模型设置标签？

Answer 1

您需要将标签编码为 [0, 4]，它是从零开始的，而不是从一开始的。这是因为为了恢复类索引，使用了 argmax 函数，该函数返回最大值对应的数组索引，该索引将从零开始。