Browse Source

gengxi (#392)

* Update README.md

* Update input_data.py

Co-authored-by: Aymeric Damien <aymeric.damien@gmail.com>
ShanksAndSS 4 years ago
parent
commit
fe8b8122f7
1 changed files with 2 additions and 2 deletions
  1. 2 2
      input_data.py

+ 2 - 2
input_data.py

@@ -90,7 +90,7 @@ class DataSet(object):
   def epochs_completed(self):
     return self._epochs_completed
   def next_batch(self, batch_size, fake_data=False):
-    """Return the next `batch_size` examples from this data set."""
+    """Return the next `batch_size`examples from this data set."""
     if fake_data:
       fake_image = [1.0 for _ in xrange(784)]
       fake_label = 0
@@ -141,4 +141,4 @@ def read_data_sets(train_dir, fake_data=False, one_hot=False):
   data_sets.train = DataSet(train_images, train_labels)
   data_sets.validation = DataSet(validation_images, validation_labels)
   data_sets.test = DataSet(test_images, test_labels)
-  return data_sets
+  return data_sets