import tensorflow as tf
import matplotlib.pyplot as plt
plt.rcParams['font.family']='SimHei'
fashion=tf.keras.datasets.fashion_mnist
(x_train,y_train),(x_test,y_test)=fashion.load_data()
x_train,x_test=x_train/255.0,x_test/255.0
plt.figure(figsize=(10,20))
dic={0:'T恤',1:'裤子',2:'套头衫',3:'连衣裙',4:'外套',5:'凉鞋',6:'衬衫',7:'运动鞋',8:'包',9:'靴子'}
for i in range(20):
plt.subplot(5,4,i+1)
plt.imshow(x_train[i])
plt.title(dic[y_train[i]])
plt.show()
model=tf.keras.models.Sequential([
tf.keras.layers.Flatten(),
tf.keras.layers.Dense(125,activation='relu'),
tf.keras.layers.Dense(100,activation='softmax')
])
model.compile(optimizer='adam',
loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=False),
metrics=['sparse_categorical_accuracy'])
model.fit(x_train,y_train,batch_size=32,epochs=200,validation_data=(x_test,y_test),validation_freq=1)
model.summary()