diff options
Diffstat (limited to 'tensorflow/contrib/learn/python/learn/datasets/mnist.py')
-rw-r--r-- | tensorflow/contrib/learn/python/learn/datasets/mnist.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/learn/python/learn/datasets/mnist.py b/tensorflow/contrib/learn/python/learn/datasets/mnist.py index 08068da182..f11e40e045 100644 --- a/tensorflow/contrib/learn/python/learn/datasets/mnist.py +++ b/tensorflow/contrib/learn/python/learn/datasets/mnist.py @@ -42,7 +42,7 @@ def extract_images(f): f: A file object that can be passed into a gzip reader. Returns: - data: A 4D unit8 numpy array [index, y, x, depth]. + data: A 4D uint8 numpy array [index, y, x, depth]. Raises: ValueError: If the bytestream does not start with 2051. @@ -81,7 +81,7 @@ def extract_labels(f, one_hot=False, num_classes=10): num_classes: Number of classes for the one hot encoding. Returns: - labels: a 1D unit8 numpy array. + labels: a 1D uint8 numpy array. Raises: ValueError: If the bystream doesn't start with 2049. |