我的模型使用以下代码编译

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

在训练过程中，我遇到了这个错误

tensorflow.python.framework.errors_impl.InvalidArgumentError: 收到的标签值为5，超出了有效范围[0, 5)。

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

回答：

你需要将你的标签编码为[0, 4]，这是基于零的编码，而不是基于一的编码。这是因为要恢复类别索引，会使用argmax函数，该函数返回最大值对应的数组索引，而该索引是基于零的。