# 数据文件 datafile = './data/data116648/mnist.json.gz'更换数据集
时间: 2024-02-19 21:00:19 浏览: 20
# 数据文件
datafile = './data/data23625/fashion-mnist.json.gz'
# 加载数据
def load_data(mode='train'):
# 读取数据文件
data_path = datafile
data_json = json.load(gzip.open(data_path))
# 读取数据
data = np.array(data_json[mode]['data']).astype(np.float32)
# 对数据做归一化处理
data = data / 255.0
# 将数据形状转换为 [batch_size, 1, 28, 28]
data = data.reshape(-1, 1, 28, 28)
# 读取标签
labels = np.array(data_json[mode]['labels']).astype(np.int64)
return data, labels
# 获取训练数据和测试数据
train_data, train_labels = load_data(mode='train')
test_data, test_labels = load_data(mode='test')