Skip to content

ebeyabraham/tsne-viz

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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