tf.keras.datasets.mnist.load_data
时间: 2023-11-21 19:02:52 浏览: 193
tf.keras.datasets
tf.keras.datasets.mnist.load_data是一个tensorflow中的函数,用于加载并返回MNIST数据集。
MNIST是一个非常经典的手写数字数据集,其中包含了60000个训练样本和10000个测试样本。该数据集被广泛用于机器学习和深度学习的训练和验证。
load_data()函数会自动下载MNIST数据集到本地,并将其分为训练集和测试集。下载完成后,load_data()函数会将训练集和测试集分别存储在两个元组中。
训练集元组包含两个NumPy数组:(x_train, y_train)。x_train包含了60000个28x28的灰度图像,每个图像代表一个手写数字样本。y_train包含了与x_train对应的60000个标签,取值范围为0到9,表示手写数字的类别。
测试集元组也包含两个NumPy数组:(x_test, y_test)。x_test包含了10000个28x28的灰度图像,每个图像代表一个手写数字样本。y_test包含了与x_test对应的10000个标签。
load_data()函数的返回结果是一个元组,其中包含了训练集元组和测试集元组。我们可以通过如下方式获取这些数据:
(x_train, y_train), (x_test, y_test) = tf.keras.datasets.mnist.load_data()
加载MNIST数据集后,我们就可以使用这些数据进行模型的训练和测试,用于识别手写数字。
阅读全文