/tsne-viz

t-SNE visualisation of CNN features

Primary LanguageJupyter NotebookMIT LicenseMIT

t-SNE Visualisation of CNN Features

t-SNE visualisation of CNN features can help understand how the CNN understands the relationship between different classes which it is trained on. This repository contains minimal code to visualise t-SNE embeddings of the features obtained after average pooling in the ResNet-18 architecture. For visualisation the validation set of Imagenette was used.

Usage

  1. The jupyter notebook generates .tsv files containing the features (vis.tsv) and the corresponding labels (metadata.csv)
  2. Upload both files to : https://projector.tensorflow.org/
  3. Et voila!

projector