Downloads · 30 days
3
33% of all-time downloads
Maksimk04/Digits_autoencoder_mnist
Digits_autoencoder_mnist is a machine learning model from Maksimk04. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for tf-keras.
Данная НС, по сути, является вариационным автоэнкодером (VAE), принимающая на вход изображение 28х28, возвращая измененное изображение той же самой цифры.
Downloads · 30 days
3
33% of all-time downloads
All-time downloads
9
Public
Repo size
3.3 MB
Likes
0
Public
Click a slice to open those files.
.data-00000-of-000013 MB · 87%
From the Hugging Face model README
Данная НС, по сути, является вариационным автоэнкодером (VAE), принимающая на вход изображение 28х28, возвращая измененное изображение той же самой цифры.
Структура модели:

Общее количество параметров составляет 249247 (124233 для энкодера и 125014 для декодера) В качестве алгоритма оптимизации был использован стандартный 'adam' из keras. Функция ошибок - mse (mean squared error). (В дальнейшем функцию ошибок лучше заменить на специальную для vae)
Размеры тренировочного и тестового датасеты стандартны: 60 тыс. тренировочный 10 тыс. тестовый В ходе обучения тренировочный разбивается еще и на валидационный в пропорции 1:5 (0.2), поэтому итоговый размер тренировочного датасета - 48 тыс., валидационный - 12 тыс.
По окончанию обучения (10 эпох): loss для тренировочной 0.334 loss для валидационной 0.335 loss для тестовой 0.336
В качестве метрики для точности для такого рода НС выбрать что-либо очень сложно,
Была выбрана стандартная метрика accuracy, которая, соответсвенно, показала не самые информативные результаты: для тренировочной 0.0092 для валидационной 0.0093 для тестовой 0.0074
Пример генерации сетью цифры 7
