Модель восстановления изображения
ИНС - автоэнкодер, на вход которой подается изображение размером 28*28. Задача ИНС - сжать изображение и восстановить его.
Общее количество обучаемых параметров НС: 4,385
Используемый алгоритм оптимизации: Adamax , функция ошибки: mse
Размеры датасетов: тренировочный - 48000 изображений, валидационный - 12000 изображений, тестовый - 10000 изображений
Результаты обучения модели на 10 эпохах:
(в качестве вычисления accuracy была использована метрика SSIM)
Тренировочный датасет: loss - 0.01716545782983303, SSIM - 0.8874326
Валидационный датасет: loss - 0.017233747988939285, SSIM - 0.8873238
Тестовый датасет: loss - 0.01724238507449627, SSIM - 0.88665247