
上QQ阅读APP看书,第一时间看更新
How to do it...
Let's look at the code on how to load MNIST data into numpy arrays using the keras.datasets.mnist class:
from keras.datasets import mnist
(X_train, y_train), (X_test, y_test) = mnist.load_data()
print("X_train shape: " + str(X_train.shape))
print("y_train shape: " + str(y_train.shape))
print("X_test shape: " + str(X_test.shape))
print("y_test shape: " + str(y_test.shape))
The output of the preceding list shows the following dataset shape:
X_train shape: (60000, 28, 28)
y_train shape: (60000,)
X_test shape: (10000, 28, 28)
y_test shape: (10000,)
Next, let's look at how to load data from a .csv file.