LSTM跑fashion-mnist（学习）

import tensorflow as tf
import tensorflow.keras as keras
from tensorflow.keras import models, layers, optimizers
import matplotlib.pyplot as plt

Mnist数据集加载

(x_train_all, y_train_all), (x_test, y_test) = keras.datasets.fashion_mnist.load_data()

Mnist数据集简单归一化

x_train_all, x_test = x_train_all / 255.0, x_test / 255.0

x_train, x_test = x_train_all[:50000], x_train_all[50000:]
y_train, y_test = y_train_all[:50000], y_train_all[50000:]
print(x_train.shape)

构建模型

inputs = layers.Input(shape=(x_train.shape[1], x_train.shape[2]), name=‘inputs’)
print(inputs.shape)
lstm = layers.LSTM(units=128, return_sequences=False)(inputs)
print(lstm.shape)
outputs = layers.Dense(10, activation=‘softmax’)(lstm)
print(outputs.shape)
lstm = keras.Model(inputs, outputs)

查看模型

lstm.compile(optimizer=keras.optimizers.Adam(0.001),
loss=‘sparse_categorical_crossentropy’,
metrics=[‘accuracy’])
lstm.summary()

训练模型

history = lstm.fit(x_train, y_train, batch_size=32, epochs=50, validation_split=0.1)