diff options
Diffstat (limited to 'tensorflow/python/keras/utils/np_utils.py')
| -rw-r--r-- | tensorflow/python/keras/utils/np_utils.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/python/keras/utils/np_utils.py b/tensorflow/python/keras/utils/np_utils.py index 9d9c72b162..c24e87308b 100644 --- a/tensorflow/python/keras/utils/np_utils.py +++ b/tensorflow/python/keras/utils/np_utils.py @@ -33,7 +33,8 @@ def to_categorical(y, num_classes=None): num_classes: total number of classes. Returns: - A binary matrix representation of the input. + A binary matrix representation of the input. The classes axis is placed + last. """ y = np.array(y, dtype='int') input_shape = y.shape |