一、导入相关库函数
import matplotlib.pyplot as plt
import tensorflow as tf
import keras
import numpy as np二、载入mnist数据集
使用keras.中的mnist数据集
(train_images, train_labels), (test_images, test_labels)=\
keras.datasets.mnist.load_data()三、测试数据的情况，数据集图像和label标签
x = train_images[2]
y = train_labels[2]
plt.title('label: %i' % y)
plt.imshow(x, cmap=plt.cm.gray_r, interpolation='nearest')
四、建立神经网络模型
keras中有API帮助建立，用Sequential的AIP建立
model = keras.Sequential([
#模型是多层的，底层是输入层，做Flatten,input_shape分辨率28*28
keras.layers.Flatten(input_shape=(28,28)),
#隐藏层，使用relu
keras.layers.Dense(128, activation=tf.nn.relu),
#输出层,10分类，数字从0~9，一共10种（选择softmax）
keras.layers.Dense(10,activation=tf.nn.softmax)
])五、将模型进行compile，优化器optimizers.Adam()，选择损失函数loss，用精度来度量
model.compile(optimizer=tf.optimizers.Adam(),
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])六、打印model，神经网络模型，三层结构
看一下神经网络模型结构：三层，输入层784，隐藏层128，输出层10
model.summary()七、训练神经网络模型，精度在增长，loss减少
epochs迭代次数，这里选择10次迭代
model.fit(train_images,train_labels,epochs=10)八、评估，测试模型性能
test_loss,test_acc = model.evaluate(test_images, test_labels)九、训练的模型进行预测
predictions = model.predict(test_images)十、测试模型，用测试集进行
预测结果为
[0. 0. 1. 0. 0. 0. 0. 0. 0. 0.]，1的index为2，预测值为2，和真实值一致，预测成功。
x_test = test_images[888]
y_test = test_labels[888]
y_pred = predictions[888]
#打印x_test图像
plt.imshow(x_test,cmap=plt.cm.gray_r,interpolation='nearest')
y_pred2 = np.around(
y_pred,
decimals=1
)
print(y_pred2)output: [0. 0. 1. 0. 0. 0. 0. 0. 0. 0.]
十一、完整代码
import matplotlib.pyplot as plt
import tensorflow as tf
import keras
import numpy as np
#载入mnist数据集
(train_images, train_labels), (test_images, test_labels)=\
keras.datasets.mnist.load_data()
#建立神经网络模型
#keras中有API帮助建立，用Sequential的AIP建立
model = keras.Sequential([
#模型是多层的，底层是输入层，做Flatten,input_shape分辨率28*28
keras.layers.Flatten(input_shape=(28,28)),
#隐藏层，使用relu
keras.layers.Dense(128, activation=tf.nn.relu),
#输出层,10分类，数字从0~9，一共10种（选择softmax）
keras.layers.Dense(10,activation=tf.nn.softmax)
])
#将模型进行compile，优化器optimizers.Adam()，选择损失函数loss，用精度来度量
model.compile(optimizer=tf.optimizers.Adam(),
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])
#训练神经网络模型，精度在增长，loss减少
#epochs迭代次数，这里选择10次迭代
model.fit(train_images,train_labels,epochs=10)
#评估，测试模型性能
#在测试数据集上进行评估
test_loss,test_acc = model.evaluate(test_images, test_labels)
#刚刚训练的模型进行预测
predictions = model.predict(test_images)
x_test = test_images[888]
y_test = test_labels[888]
y_pred = predictions[888]
#打印x_test图像
plt.imshow(x_test,cmap=plt.cm.gray_r,interpolation='nearest')
y_pred2 = np.around(
y_pred,
decimals=1
)
print(y_pred2)
