我一直在尝试构建一个神经网络来训练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)