Ошибка индекса - Python - Набор данных EMNIST
Я пытался построить нейронную сеть для обучения наборов данных EMNIST. Два сегмента кода ниже находятся в совершенно разных ячейках в блокноте jupyter, однако именно они и являются причиной ошибки, указанной ниже. Моя проблема связана с тем, что для одного набора данных код работает нормально, а для этого конкретного набора данных я получаю сообщение об ошибке. Если кто-нибудь может сказать мне, где я иду не так, это было бы очень признательно.
IndexError: index 540774 is out of bounds for size 540774
def dense_to_one_hot(labels_dense, num_classes):
num_labels = labels_dense.shape[0]
index_offset = np.arange(num_labels) * num_classes
labels_one_hot = np.zeros((num_labels, num_classes))
labels_one_hot.flat[index_offset + labels_dense.ravel()] = 1
return labels_one_hot
test_labels_flat = test_data_labels[["1"]].values.ravel()
test_labels_count = np.unique(test_labels_flat).shape[0]
test_labels = dense_to_one_hot(test_labels_flat, test_labels_count)
test_labels = test_labels.astype(np.uint8)