Numpy 不理解从读取h5数据集文件中的类零件和重塑

您好，有人能一步一步地解释一下下面的代码是什么吗？ 特别是零件类别和重塑？tnx

def load_data():
    train_dataset = h5py.File('datasets/train_catvnoncat.h5', "r")
    train_set_x_orig = np.array(train_dataset["train_set_x"][:]) # your train set features
    train_set_y_orig = np.array(train_dataset["train_set_y"][:]) # your train set labels

    test_dataset = h5py.File('datasets/test_catvnoncat.h5', "r")
    test_set_x_orig = np.array(test_dataset["test_set_x"][:]) # your test set features
    test_set_y_orig = np.array(test_dataset["test_set_y"][:]) # your test set labels

    classes = np.array(test_dataset["list_classes"][:]) # the list of classes

    train_set_y_orig = train_set_y_orig.reshape((1, train_set_y_orig.shape[0]))
    test_set_y_orig = test_set_y_orig.reshape((1, test_set_y_orig.shape[0]))

    return train_set_x_orig, train_set_y_orig, test_set_x_orig, test_set_y_orig, classes

---

大多数行只是从

h5

文件加载

数据集。不需要
np.array（…）
包装器<代码>测试数据集[名称][：]

足以加载数组

test_set_y_orig = test_dataset["test_set_y"][:]

test\u数据集

是打开的文件<代码>测试数据集[“测试集”]是该文件上的

数据集。
[：]
将数据集加载到
numpy
数组中。查看
h5py
文档，了解有关加载
数据集的更多详细信息

我从

train_set_y_orig = train_set_y_orig.reshape((1, train_set_y_orig.shape[0]))

加载时的数组是1d，形状为
（n，）
，而此重塑只是添加了一个初始维度，使其成为
（1，n）
。我会把它编码为

train_set_y_orig = train_set_y_orig[None,:]

但结果是一样的

类
数组没有什么特别之处（尽管它很可能是一个字符串数组）