Как определить значение пользовательских меток при использовании sparse_categorical_crossentropy?

Question

моя модель скомпилирована с этим кодом

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

Во время обучения я получил эту ошибку

ensororflow.python.framework.errors_impl.InvalidArgumentError: получил значение метки5, который находится за пределами допустимого диапазона [0, 5).

Моя метка 1,2,3,4,5, которая [1,5] не [0, 5).Как я могу установить метки для этой модели?

Answer 1

Вам необходимо закодировать ваши метки в [0, 4], который начинается с нуля, а не с единицы.Это связано с тем, что для восстановления индексов классов используется функция argmax, для которой возвращается индекс массива, соответствующий максимальному значению, который будет начинаться с нуля.