引言
在深度学习的世界里,MNIST数据集相当于是“Hello World”程序。它包含了大量的手写数字图像,是初学者学习图像识别和训练神经网络的理想起点。在这篇博客中,我们将结合NVIDIA深度学习DLI基础课程的内容,学习如何使用MNIST数据集来训练我们的第一个深度学习模型。
MNIST数据集的魅力
MNIST数据集由60,000个训练图像和10,000个测试图像组成,每个图像都是一个28x28像素的手写数字。这些图像被标记为0到9的数字,为我们提供了一个清晰的分类任务。
准备数据
在开始之前,我们需要对数据进行一些基本的预处理。这包括将图像数据展平为一维数组、归一化像素值,以及对标签进行多分类编码。
from tensorflow.keras.datasets import mnist
import tensorflow.keras as keras
# 加载MNIST数据集
(x_train, y_train), (x_test, y_test) = mnist.load_data()
# 展平图像数据
x_train = x_train.reshape(60000, 784)
x_test = x_test.reshape(10000, 784)
# 归一化图像数据
x_train = x_train / 255
x_test = x_test / 255
# 对标签进行多分类编码
num_categories = 10
y_train = keras.utils.to_categorical(y_train, num_categories)
y_test = keras.utils.to_categorical(y_test, num_categories)
创建和编译模型
接下来,我们将使用Keras的Sequential
模型类来创建我们的神经网络。这个网络将包含一个输入层、两个隐藏层和一个输出层。
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense
model = Sequential()
# 创建输入层和第一个隐藏层
model.add(Dense(units=512, activation='relu', input_shape=(784,)))
# 创建第二个隐藏层
model.add(Dense(units=512, activation='relu'))
# 创建输出层
model.add(Dense(units=10, activation='softmax'))
在模型创建好后,我们需要对其进行编译,指定损失函数和优化器。
model.compile(loss='categorical_crossentropy', metrics=['accuracy'])
训练模型
现在,我们可以使用训练数据来训练模型,并监控其在测试数据上的表现。
history = model.fit(x_train, y_train,
epochs=5,
verbose=1,
validation_data=(x_test, y_test))
观察和理解结果
训练完成后,我们可以查看模型的准确率,并分析其在训练和验证数据上的表现。通常,我们希望看到准确率随着时间提高,并在训练和验证数据上都达到较高的水平。
结语
通过这篇博客,我们不仅学会了如何处理数据、构建神经网络、训练模型,还对深度学习在图像识别领域的应用有了基本的了解。MNIST数据集作为一个起点,为我们打开了深入探索深度学习世界的大门。