loading data of mnist

LordSomen · Jul 30, 2018 · 33cceca · 33cceca
1 parent 66b5197
commit 33cceca
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/Ensemble_learning/mnist_ensemble.py b/Ensemble_learning/mnist_ensemble.py
@@ -0,0 +1,19 @@
+#%%
+''' retrieving mnist data '''
+
+from sklearn.datasets import fetch_mldata
+mnist = fetch_mldata('MNIST original')
+print(mnist)
+
+#%%
+X, Y = mnist["data"], mnist["target"]
+print(X)
+print(Y)
+print(X.shape)
+print(Y.shape)
+
+#%%
+X_train,Y_train,X_val,Y_val,X_test,Y_test = X[:40000],Y[:40000],X[40000:50000],Y[40000:50000],X[50000:],Y[50000:]
+print(X_val)
+
+#%%