tf.keras.utils.GeneratorEnqueuer

Builds a queue out of a data generator.

Compat aliases for migration

tf.compat.v1.keras.utils.GeneratorEnqueuer

tf.keras.utils.GeneratorEnqueuer(
    generator, use_multiprocessing=False, random_seed=None
)

The provided generator can be finite in which case the class will throw a StopIteration exception.

Args
`generator`	a generator function which yields data
`use_multiprocessing`	use multiprocessing if True, otherwise threading
`random_seed`	Initial seed for workers, will be incremented by one for each worker.

Methods

get()

Creates a generator to extract data from the queue.

Skip the data if it is None.

Yields
The next element in the queue, i.e. a tuple `(inputs, targets)` or `(inputs, targets, sample_weights)`.

is_running()

start(
    workers=1, max_queue_size=10
)

Starts the handler's workers.

Args
`workers`	Number of workers.
`max_queue_size`	queue size (when full, workers could block on `put()`)

stop(
    timeout=None
)

Stops running threads and wait for them to exit, if necessary.

Should be called by the same thread which called start().

Args
`timeout`	maximum time to wait on `thread.join()`

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/keras/utils/GeneratorEnqueuer