Animation of tSNE on 2500 MNIST digits

Check out this tSNE animation. It runs slowly because heroku has no gpu support. If you run the code on a machine with pytorch and a gpu, it is much more satisfying.

This was a dual-purpose project:

  • To port tSNE to pytorch
  • To write code for animations in bokeh using the various examples in the bokeh documentation on asynchronously updating plots.

The code
is based on the Python implementation given on van der Maaten’s web page.

Here is an screencast of the app running with gpu support:


@jeremy9959 this looks neat! If you ever have a chance to make a screen grab of the animation running on a GPU it would be a great addition to the post!

I went ahead and added a screencast.

1 Like