手动实现 Autoencoder(基于Keras实现)
download the file from:
#“https://s3.amazonaws.com/img-datasets/mnist.npz”
import tensorflow as tf
import numpy as np
(x_train,y_train),(x_test,y_test)= load_data(‘MNIST_DATA/mnist.npz’)
x_train,x_test = x_train/255,x_test/255
model = tf.keras.models.Sequential([
tf.keras.layers.Flatten(),
tf.keras.layers.Dense(512,activation = tf.nn.relu),
tf.keras.layers.Dense(38,activation = tf.nn.relu),
tf.keras.layers.Dense(512,activation = tf.nn.relu),
tf.keras.layers.Dense(784,activation = tf.nn.relu),
tf.keras.layers.Dense(10,activation = tf.nn.sigmoid)
])
model.compile(optimizer='adam',
loss = 'sparse_categorical_crossentropy',
matrics=['accuracy'])
model.fit(x_train,y_train,epochs=200)
model.evaluate(x_test,y_test)