Keras 是一个高层神经网络 API,它能够以Tensorflow, CNTK, 或者 Theano 作为后端运行。TensorFlow 是一个开源的机器学习库,用于数据流图中的数值计算。MNIST 数据集是一个手写数字的数据集,包含了60000个训练样本和10000个测试样本,每个样本都是28x28像素的灰度图像。
使用 Keras 和 TensorFlow 下载 MNIST 数据集非常简单。以下是一段示例代码:
import tensorflow as tf
from tensorflow.keras.datasets import mnist
# 下载并加载 MNIST 数据集
(x_train, y_train), (x_test, y_test) = mnist.load_data()
# 打印数据集的形状
print("训练数据形状:", x_train.shape)
print("训练标签形状:", y_train.shape)
print("测试数据形状:", x_test.shape)
print("测试标签形状:", y_test.shape)
如果在下载 MNIST 数据集时遇到问题,可能是由于网络问题导致的下载失败。可以尝试以下方法解决:
import numpy as np
from tensorflow.keras.utils import get_file
# 手动下载的 MNIST 数据集路径
path = "path_to_downloaded_mnist.npz"
# 加载数据集
with np.load(path) as f:
x_train, y_train = f['x_train'], f['y_train']
x_test, y_test = f['x_test'], f['y_test']
# 打印数据集的形状
print("训练数据形状:", x_train.shape)
print("训练标签形状:", y_train.shape)
print("测试数据形状:", x_test.shape)
print("测试标签形状:", y_test.shape)
请注意,以上代码和信息是基于当前的库版本,如果库有更新,可能需要调整代码以适应新的API。
领取专属 10元无门槛券
手把手带您无忧上云