这是官网的一个入门案例,识别手写数1-9。
加载mnist数据集
import tensorflow as tf minist=tf.keras.datasets.mnist (x_train,y_train),(x_test,y_test)=minist.load_data()
然后会报一个错误
load_data是去谷歌的服务器上下载数据,所以不能开学上网的话貌似是连不上的。
可以去网上找一个mnist.npz下载到本地。
然后改写为
path='C:/Users/.../Documents/PY code/mnist.npz' (x_train,y_train),(x_test,y_test)=minist.load_data(path)
就可以了。
import tensorflow as tf minist=tf.keras.datasets.mnist path='C:/Users/.../Documents/PY code/mnist.npz' (x_train,y_train),(x_test,y_test)=minist.load_data(path) x_train,x_test=x_train/255.0,x_test/255.0 #将神经网络模型各层堆叠起来,为训练选择优化器和损失函数。 model=tf.keras.Sequential([ tf.keras.layers.Flatten(input_shape=(28,28)), tf.keras.layers.Dense(128,activation='relu'), tf.keras.layers.Dropout(0.2), #退出率,一般在后几层防止过拟合 tf.keras.layers.Dense(10,activation='softmax') ]) model.compile(optimizer='adam',#优化器,自适应矩估计,每一代学习率会有变化 loss='sparse_categorical_crossentropy',#稀疏分类交叉熵 metrics=['accuracy'] #评估标准,准确度 ) model.fit(x_train,y_train,epochs=5) model.evaluate(x_test,y_test,verbose=2)#