from sklearn.datasets import fetch_openml mnist = fetch_openml('mnist_784')
In [6]:
X, y =mnist['data'], mnist['target'] X.shape
Out[6]:
(70000, 784)
In [10]:
y.shape X.index
Out[10]:
RangeIndex(start=0, stop=70000, step=1)
In [30]:
%matplotlib inline import matplotlib import matplotlib.pyplot as plt some_digit = X.loc[36011].values print(type(some_digit)) some_digit_image = some_digit.reshape(28, 28) plt.imshow(some_digit_image, cmap = matplotlib.cm.binary, interpolation="nearest") plt.axis("off") plt.show()
<class 'numpy.ndarray'>
In [32]:
y[36011]
Out[32]:
'5'
In [45]:
X_train,