Update the shape of y_train and y_test
Update the shape of y_train and y_test in the output of CIFAR10 and CIFAR100 datasets.
@@ -15,7 +15,7 @@ from keras.datasets import cifar10
- __Returns:__
- 2 tuples:
- __x_train, x_test__: uint8 array of RGB image data with shape (num_samples, 3, 32, 32) or (num_samples, 32, 32, 3) based on the `image_data_format` backend setting of either `channels_first` or `channels_last` respectively.
- - __y_train, y_test__: uint8 array of category labels (integers in range 0-9) with shape (num_samples,).
+ - __y_train, y_test__: uint8 array of category labels (integers in range 0-9) with shape (num_samples, 1).
---
@@ -35,7 +35,7 @@ from keras.datasets import cifar100
- __Returns:__
- 2 tuples:
- __x_train, x_test__: uint8 array of RGB image data with shape (num_samples, 3, 32, 32) or (num_samples, 32, 32, 3) based on the `image_data_format` backend setting of either `channels_first` or `channels_last` respectively.
- - __y_train, y_test__: uint8 array of category labels with shape (num_samples,).
+ - __y_train, y_test__: uint8 array of category labels with shape (num_samples, 1).
- __Arguments:__
