PyTorch-tSNE

You can visualize high dimensional data using t-SNE. This t-SNE implementation code uses TSNE from scikit-learn and it might take a little bit long time to generate the image. This code can be easily applied to other datasets such as CIFAR10, CIFAR100, etc.

Usage

python3 main.py

STL10 example

This example image is generated by ImageNet pretrained ResNet18 on STL10 dataset.