A PyTorch implementation of 'Dimensionality Reduction by Learning an Invariant Mapping' which has published on CVPR (2006).
- python 3
- pytorch 1.0
- einops
- To train the model, execute
python train.py
- To test the model and plot the clusters after dimensionality reduction, execute
python plot.py
Result on MNIST: