tensorflow(9)--识别FASHION数据集

tensorflow(9)--识别FASHION数据集

代码如下

import tensorflow as tf
from tensorflow import keras
import matplotlib.pyplot as plt

fashion = 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


model = keras.models.Sequential([
    keras.layers.Flatten(),
    keras.layers.Dense(128,activation='relu'),
    keras.layers.Dense(10,activation="softmax")
])

model.compile(optimizer='adam',loss=keras.losses.SparseCategoricalCrossentropy(from_logits=False),metrics=['sparse_categorical_accuracy'])
model.fit(x_train,y_train,batch_size=32,epochs=5,validation_data=(x_test,y_test),validation_freq=1)
model.summary()

刚写完上一篇文章有点小累，水一篇哈，这篇的代码在上一篇都有讲解，不懂的回去瞅瞅，复习一下，俺溜了，下篇好好写，下篇见嗷小伙伴们！0^0