|
@@ -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
|