Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
datasets:
|
3 |
+
- mnist
|
4 |
+
---
|
5 |
+
|
6 |
+
Данная НС, по сути, является вариационным автоэнкодером (VAE), принимающая на вход изображение 28х28,
|
7 |
+
возвращая измененное изображение той же самой цифры.
|
8 |
+
|
9 |
+
Структура модели:
|
10 |
+
![]()
|
11 |
+
|
12 |
+
Общее количество параметров составляет 249247 (124233 для энкодера и 125014 для декодера)
|
13 |
+
В качестве алгоритма оптимизации был использован стандартный 'adam' из keras.
|
14 |
+
Функция ошибок - mse (mean squared error).
|
15 |
+
(В дальнейшем функцию ошибок лучше заменить на специальную для vae)
|
16 |
+
|
17 |
+
Размеры тренировочного и тестового датасеты стандартны:
|
18 |
+
60 тыс. тренировочный
|
19 |
+
10 тыс. тестовый
|
20 |
+
В ходе обучения тренировочный разбивается еще и на валидационный в пропорции 1:5 (0.2),
|
21 |
+
поэтому итоговый размер тренировочного датасета - 48 тыс., валидационный - 12 тыс.
|
22 |
+
|
23 |
+
По окончанию обучения (10 эпох):
|
24 |
+
loss для тренировочной 0.334
|
25 |
+
loss для валидационной 0.335
|
26 |
+
loss для тестовой 0.336
|
27 |
+
|
28 |
+
В качестве метрики для точности для такого рода НС выбрать что-либо очень сложно,
|
29 |
+
|
30 |
+
Была выбрана стандартная метрика accuracy,
|
31 |
+
которая, соответсвенно, показала не самые информативные результаты:
|
32 |
+
для тренировочной 0.0092
|
33 |
+
для валидационной 0.0093
|
34 |
+
для тестовой 0.0074
|