diff --git a/docs/templates/datasets.md b/docs/templates/datasets.md index a2f1d1a1eabc..a35483943796 100644 --- a/docs/templates/datasets.md +++ b/docs/templates/datasets.md @@ -14,7 +14,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). + - __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,). @@ -34,7 +34,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). + - __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,). - __Arguments:__ @@ -206,4 +206,4 @@ from keras.datasets import boston_housing - __test_split__: fraction of the data to reserve as test set. - __Returns:__ - Tuple of Numpy arrays: `(x_train, y_train), (x_test, y_test)`. \ No newline at end of file + Tuple of Numpy arrays: `(x_train, y_train), (x_test, y_test)`.