/Contractive_Autoencoder_in_Pytorch

Pytorch implementation of contractive autoencoder on MNIST dataset

Primary LanguagePython

Contractive_Autoencoder_in_Pytorch

Pytorch implementation of contractive autoencoder on MNIST dataset.

To run this code just type the following in your terminal: python CAE_pytorch.py

Result:

alt text

Requirements:

(i) PyTorch (ii) Python 3.6 (iii) matplotlib

Citation:

Rifai, Salah, et al. “Contractive auto-encoders: Explicit invariance during feature extraction.” Proceedings of the 28th international conference on machine learning (ICML-11). 2011.