Простая нейросеть, состоящая из трёх слоёв, решающая задачу распознавания цифр. Обучена и протестирована на датасете SVHN. Точность на тестовых данных - порядка 85%.
Эскиз структуры нейронной сети.
Нейросеть реализована с использованием фреймворка pyTorch.
Файл main.py
содержит в себе весь цикл обучения модели, остальные файлы хранят в себе вспомогательные функции, помещенные отдельно для компактности.
Файл Notebook.ipynb
позволяет удобно запустить тренировку нейронной сети в одном из входов ноутбука. В этом файле также представлены функции для визуализации точности и функции ошибки во время обучения.
Вдохновение и основу для этого решения взял из домашних заданий с курса лекций dlcourse.ai.