autoencoder / README.md
jxssx's picture
Update README.md
bd2defc
Данная нейронная сеть восстанавливает входное изображение из "скрытого" состояния. Таким образом, на выходе получается новое изображение.
![](model.png)
Алгоритм оптимизации: Adam.
Функция ошибки выглядит так:
def loss(y, z):
y = K.reshape(y, shape = (batch_size, 28*28))
z = K.reshape(z, shape = (batch_size, 28*28))
mse = K.sum(K.square(y - z), axis = 1)
kl = -.5 * K.sum(1 + loss_z_log_var - K.square(loss_z_mean) - K.exp(loss_z_log_var), axis = 1)
return mse
Длина тренировочного и тестового датасетов: 60000 и 10000 соответственно.
Потери в процессе обучения:
![](loss.png)