In [1]:
!wget http://deeplearning.net/data/mnist/mnist.pkl.gz


--2013-08-25 19:00:43--  http://deeplearning.net/data/mnist/mnist.pkl.gz
Resolving deeplearning.net... 132.204.24.80
Connecting to deeplearning.net|132.204.24.80|:80... connected.
HTTP request sent, awaiting response... 200 OK
Length: 16168813 (15M) [application/x-gzip]
Saving to: ‘mnist.pkl.gz’

100%[======================================>] 16,168,813   780KB/s   in 26s    

2013-08-25 19:01:10 (614 KB/s) - ‘mnist.pkl.gz’ saved [16168813/16168813]


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]:
<matplotlib.collections.PathCollection at 0x10970b650>

In [ ]: