mnist手写体识别

import keras
from keras import layers
import matplotlib.pyplot as plt

import keras.datasets.mnist as minst

(train_image,train_label),(test_image,test_label) = minst.load_data()

print(train_image.shape)
plt.imshow(train_image[1000])
plt.show()
print(train_label[1000])
print(test_image.shape,test_label)

model = keras.Sequential()
model.add(layers.Flatten()) # 从(60000,28,28)-->（6000，28*28）
model.add(layers.Dense(64,activation='relu'))
model.add(layers.Dense(64,activation='relu'))
model.add(layers.Dense(64,activation='relu'))
model.add(layers.Dense(64,activation='relu'))
model.add(layers.Dense(10,activation='softmax'))

model.compile(optimizer='adam',
              loss = 'sparse_categorical_crossentropy',
              metrics =['acc'])

model.fit(train_image,train_label,epochs=50,batch_size=512) #分批次训练训练，不然数据量太大，容易爆存，同时批次尽可能大，效果比较好
# model.evaluate(train_image,train_label)

使用的是Keras编写的