AxisError：轴 1 超出了维度 1 数组的边界 （OneHotEncoded python）

问：

我正在研究犬种的分类模型，我试图显示标签的示例及其各自的一个热编码标签，但我收到一个错误，说 AxisError：轴 1 超出了维度 1 数组的边界。

错误信息

这是我当前的代码：

    def decode_one_hot(one_hot_encoded, labels):
        # Use numpy's argmax to get the index of the '1' in each encoded list
        indices = np.argmax(one_hot_encoded, axis=1)
    
        # Convert indices back to original labels
        decoded_labels = [labels[index] for index in indices]
        return decoded_labels

    print('Example of label and one_hot_encoded label')
    train_labels = decode_one_hot(y_train, breeds)
    show_images(image_array= X_train, labels=train_labels, encoded_labels=y_train)

这是我期望实现的目标： 预期结果