/TriMap-PyTorch

Implementation of TriMap dimensionality reduction in PyTorch

Primary LanguagePythonApache License 2.0Apache-2.0

TriMap-PyTorch

Implementation of TriMap dimensionality reduction in PyTorch

Making .gif animations requires imagemagick.

cd data
python3 mnist_np.py
cd ..
python3 main.py --dataset=mnist70k --num_iters=1500 --save_fig --animate --verbose

Animations of training (see tSNE-Animation for the equivalent animations with t-SNE):

batch gradient descent with momentum

linear annealing

Authors

Keller Jordan, Ehsan Amid