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)