图像分类异常给定标签的InvalidArgumentError

Question

我正在做皮肤癌图像分类的研究。我的数据集由三个类的图像组成。我给了每个类0，1和2的标签。 0-一级图像 1=二级图像 2=三等图像 当我在编译我的模型时，我得到了一个错误，这是真正困扰我。我得到的错误是-

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

我用来编译模型的代码是-

model.compile(loss='sparse_categorical_crossentropy',optimizer = optimizers.RMSprop(lr=2e-5),metrics=['acc'])

有人能告诉我我的错误吗？？在

EDIT 1 : my model code is -

^{pr2}$

Answer 1

模型的输出只对应一个类，而不是三个类，因此应该替换以下行：

model.add(layers.Dense(3,activation='softmax'))