10、构建首个深度学习 Web 应用程序

构建首个深度学习 Web 应用程序

1. 读取标签文件函数的创建

MNIST 数据集中有两个标签文件： train-labels-idx1-ubyte 和 t10k-labels-idx1-ubyte 。可以使用以下函数查看这些文件，该函数以文件名作为输入参数，输出一个独热编码标签数组：

def loadLabelFile(filelabel):
    f = open(filelabel, "rb")
    f.read(8)
    labels_arr = []
    while True:
        row = [0 for x in range(10)]
        try:
            label = ord(f.read(1))
            row[label] = 1
            labels_arr.append(row)
        except:
            break
    f.close()
    label_sets = np.array(labels_arr)
    return label_sets

此函数返回一个独热编码的 NumPy 标签数组，其维度为数据集中样本数量乘以 10。为了理解独热编码的本质，我们观察一个单独的条目。运行以下代码，该代码实际上打印了测试集中第一个样本的独热编码标签集：

test_labels = loadLabelFile("t