In [1]:
!wget http://deeplearning.net/data/mnist/mnist.pkl.gz
In [2]:
import numpy as np
import gzip, cPickle
from tsne import bh_sne
In [3]:
f = gzip.open("mnist.pkl.gz", "rb")
train, val, test = cPickle.load(f)
f.close()
In [4]:
X = np.asarray(np.vstack((train[0], val[0], test[0])), dtype=np.float64)
In [5]:
y = np.hstack((train[1], val[1], test[1]))
In [6]:
X_2d = bh_sne(X)
In [7]:
rcParams['figure.figsize'] = 20, 20
In [8]:
scatter(X_2d[:, 0], X_2d[:, 1], c=y)
Out[8]:
In [ ]: