Create README.md
Browse files![Архитектура.png](https://s3.amazonaws.com/moonup/production/uploads/647f7020e9c81260ff881661/4IfjN_BKYtmfQoum6fj_Z.png)
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 |
+
```
|