Contractive_Autoencoder_in_Pytorch
Contractive_Autoencoder_in_Pytorch copied to clipboard
Pytorch implementation of contractive autoencoder on MNIST dataset
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:

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.