Keras Deep Learning Cookbook
上QQ阅读APP看书,第一时间看更新

How to do it...

Let's look at how to load this dataset and print the shapes for X_train, y_train, X_test, and y_test. The CIFAR-100 dataset is available through the load_data() function in keras.datasets.cifar100.

The dataset is downloaded from https://www.cs.toronto.edu/~kriz/cifar-100-python.tar.gz; this is hidden in the following implementation:

from keras.datasets import cifar100
(X_train, y_train), (X_test, y_test) = cifar100.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 listing sizes is shown in the following snippet:

X_train shape: (50000, 32, 32, 3)
y_train shape: (50000, 1)
X_test shape: (10000, 32, 32, 3)
y_test shape: (10000, 1)