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