| | |
| | """Zadanie4_Semenov_II_DRPK47.ipynb |
| | |
| | Automatically generated by Colaboratory. |
| | |
| | Original file is located at |
| | https://colab.research.google.com/drive/11fNvzrVniDSjEVdE-ZnejpPvvA88ApfY |
| | """ |
| |
|
| | import numpy as np |
| | import matplotlib.pyplot as plt |
| | import tensorflow.keras as keras |
| | import tensorflow.keras.datasets |
| | from tensorflow.keras.datasets import fashion_mnist |
| | from tensorflow.keras.layers import Input, Dense |
| |
|
| | (train_x, train_y), (test_x, test_y) = fashion_mnist.load_data() |
| |
|
| | train_x = train_x / 255 |
| | test_x = test_x / 255 |
| |
|
| | train_x = np.reshape(train_x, (len(train_x), 28 * 28)) |
| | test_x = np.reshape(test_x, (len(test_x), 28 * 28)) |
| |
|
| | inputs = Input(shape = (28*28, )) |
| | x = Dense(150, activation = 'relu')(inputs) |
| | x = Dense(400, activation = 'relu')(x) |
| | x = Dense(10, activation = 'relu')(x) |
| | encoder = Dense(3, activation = 'linear')(x) |
| |
|
| |
|
| | inputs_dec = Input(shape = (3, )) |
| | x = Dense(10, activation = 'relu')(inputs_dec) |
| | x = Dense(40, activation = 'relu')(x) |
| | x = Dense(150, activation = 'relu')(x) |
| | decoder = Dense(28*28, activation = 'relu')(x) |
| |
|
| | encoder_model = keras.Model(inputs, encoder) |
| | decoder_model = keras.Model(inputs_dec, decoder) |
| | autoenc = keras.Model(inputs, decoder_model(encoder_model(inputs))) |
| |
|
| | autoenc.compile(optimizer='adam', loss='mean_squared_error', metrics = ['accuracy']) |
| |
|
| | autoenc.fit(train_x, train_x, epochs = 20, batch_size=50) |
| |
|
| | y = autoenc.predict(test_x[:12]) |
| | plt.imshow(y[5].reshape(28, 28), cmap = 'gray') |
