Maksimk04 commited on
Commit
ee264c1
·
1 Parent(s): 96851ce

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +34 -0
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