README.md

Разреженные автокодировщики

Установка

pip3 install sparse_ae/

Для установки tensorflow-gpu замените tensorflow на tensorflow-gpuв файле requirements.txt

Использование

python train_*.py - обучение автокодировщика.

python compare_models.py - строит гистограмму активаций обученных моделей.

plot_latent_code.py - восстановление изображений из латентного представления.

partial_reconstruction.py - восстановление изображений по top-k активных нейроно латентного слоя.

Файлы

• model.py - создание моделей, KL-регуляризатор
• utils.py - утилиты для подготовки данных и создания изображений.