Skip to main content
100%

A notebook describing how to use t-SNE to visualize a neural network learn representations

✓ Published0🌍 Public
Aalvinthai
Last edited Mar 2, 2016
Created on Feb 25, 2017

This notebook demonstrates how a convolutional neural network learns to recognize handwritten digits from the MNIST database by visualizing its internal representations with t-SNE. It shows the network’s high-level feature layer (h_fc2) projected into two dimensions at various training stages, revealing clusters that correspond to digit classes. The code trains a LeNet-style model using TensorFlow, then applies sklearn.manifold.TSNE to the learned activations. Matplotlib renders the resulting scatter plots, with each point colored by its true label, illustrating how representations become more separable as training progresses.

AI-generated description

Similar vizzes