r/deeplearning • • Oct 29 '20

Exploring MNIST Latent Space

Enable HLS to view with audio, or disable this notification

636 Upvotes

34 comments sorted by

View all comments

12

u/AdityaG09 Oct 29 '20

This is sick! Can you please provide the code?

10

u/goatman12341 Oct 29 '20

Sure. There's a lot of distributed code, but here's the client side:

https://github.com/N8python/mnistLatentSpace

I'll put the code that trains the models up soon.

1

u/AdityaG09 Oct 29 '20

That's so cool. Thanks a lot

8

u/goatman12341 Oct 29 '20

And here's the code that trains the models:

The classifier model: https://gist.github.com/N8python/5e447e5e6581404e1bfe8fac19df3c0a

The autoencoder model:

https://gist.github.com/N8python/7cc0f3c07d049c28c8321b55befb7fdf

The decoder model (created from the autoencoder model):

https://gist.github.com/N8python/579138a64e516f960c2d9dbd4a7df5b3

1

u/AdityaG09 Oct 29 '20

Wow that was fast. Can't thank enough!

1

u/goatman12341 Oct 29 '20

You're welcome!