В проекте реализуется автоэнкодер, кодирующий координаты точек (x,y)
на единичной окружности в латентное пространство единичной размерности. В ходе работы был написан код на PyTorch
, с помощью которого была подобрана архитертура модели, а так же была написана реализация на numpy
.
- В папке
/src
находятся вспомогательные скрипты с генерацией данных, загрузчиками данных, построением графиков, а также скрипты с реализацией модели и ее обучение наPyTorch
иnumpy
. - В папке
/notebooks
находятся ноутбуки с примерами обучения и работы обоих реализаций, с подбором архитектуры и параметров обучения наPyTorch
, подбором параметров наnumpy
.