tf.keras.utils.GeneratorEnqueuer
| Просмотреть исходный код на GitHub |
Создаёт очередь из генератора данных.
Наследуется от: SequenceEnqueuer
tf.keras.utils.GeneratorEnqueuer(
sequence, use_multiprocessing=False, random_seed=None
)
Предоставленный генератор может быть конечным, в этом случае класс будет генерировать исключение StopIteration.
Используется в fit_generator, evaluate_generator, predict_generator.
| Аргументы | |
|---|---|
generator | функция-генератор, которая возвращает данные |
use_multiprocessing | использовать многопоточность, если True, иначе — потоки |
wait_time | время ожидания между вызовами put() |
random_seed | начальное значение для генераторов, увеличивается на единицу для каждого генератора. |
Методы
get
get()
Создаёт генератор для извлечения данных из очереди.
Пропускает данные, если они None.
| Возвращает | |
|---|---|
Следующий элемент в очереди, т.е. кортеж (inputs, targets) или (inputs, targets, sample_weights). |
is_running
is_running()
start
start(
workers=1, max_queue_size=10
)
Запускает обработчики генераторов.
| Аргументы | |
|---|---|
workers | Количество обработчиков. |
max_queue_size | Размер очереди (при заполнении обработчики могут заблокироваться на put()) |
stop
stop(
timeout=None
)
Останавливает работающие потоки и ждёт их завершения, если необходимо.
Должен вызываться тем же потоком, который вызвал start().
| Аргументы | |
|---|---|
timeout | максимальное время ожидания thread.join() |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/keras/utils/GeneratorEnqueuer
