YukiNon commited on
Commit
57c902a
1 Parent(s): 2d15ff4

Create README.md

Browse files

![Архитектура.png](https://s3.amazonaws.com/moonup/production/uploads/647f7020e9c81260ff881661/4IfjN_BKYtmfQoum6fj_Z.png)

Files changed (1) hide show
  1. README.md +38 -0
README.md ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ```
2
+ ---
3
+ tags:
4
+ - autoencoder
5
+ - fashion-mnist
6
+ - pytorch
7
+ - huggingface
8
+
9
+ ---
10
+
11
+ # Энкодер на датасете fashion_mnist
12
+
13
+ ## Задача
14
+ Построить автоэнкодер для датасета Fashion MNIST, который будет принимать на вход изображение и создавать его же изображение на выходе.
15
+
16
+ ## Архитектура
17
+ ![Архитектура.png](https://s3.amazonaws.com/moonup/production/uploads/647f7020e9c81260ff881661/4IfjN_BKYtmfQoum6fj_Z.png)
18
+
19
+ ## Общее количество обучаемых параметров
20
+ Автоэнкодер содержит 235,978 обучаемых параметров
21
+
22
+ ## Функция оптимизации и функция ошибок
23
+ Алгоритм оптимизации - Adam, функция ошибок - бинарная кросс-энтропия.
24
+
25
+ ## Размеры датасета
26
+ - Тренировочный датасет: 48,000 изображений
27
+ - Валидационный датасет: 16,000 изображений
28
+ - Тестовый датасет: 16,000 изображений
29
+
30
+ ## Результаты обучения модели на всех трех датасетах
31
+ После 50 эпох обучения модели на тренировочном датасете, результаты на трех датасетах такие:
32
+ - Train Loss: 0.26351
33
+ - Train Accuracy: 0.27152
34
+ - Val Loss: 0.26502
35
+ - Val Accuracy: 0.26352
36
+ - Test Loss: 0.26442
37
+ - Test Accuracy:0.26600
38
+ ```