diff options
Diffstat (limited to 'tensorflow/contrib/learn/python/learn/learn_io/generator_io.py')
-rw-r--r-- | tensorflow/contrib/learn/python/learn/learn_io/generator_io.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/contrib/learn/python/learn/learn_io/generator_io.py b/tensorflow/contrib/learn/python/learn/learn_io/generator_io.py index 5859bb6b47..c302c7725a 100644 --- a/tensorflow/contrib/learn/python/learn/learn_io/generator_io.py +++ b/tensorflow/contrib/learn/python/learn/learn_io/generator_io.py @@ -18,8 +18,9 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function -from types import FunctionType, GeneratorType from collections import Container +from types import FunctionType +from types import GeneratorType from tensorflow.contrib.learn.python.learn.dataframe.queues import feeding_functions @@ -33,7 +34,7 @@ def generator_input_fn(x, num_threads=1): """Returns input function that would dicts of numpy arrays yielded from a generator. - + It is assumed that every dict yielded from the dictionary represents a single sample. The generator should consume a single epoch of the data. |