tf.keras.utils.SequenceEnqueuer

Базовый класс для очереди ввода.

tf.keras.utils.SequenceEnqueuer(
    sequence, use_multiprocessing=False
)

Задача Enqueuer — использовать параллелизм для ускорения предобработки. Это делается с помощью процессов или потоков.

Пример:

enqueuer = SequenceEnqueuer(...)
enqueuer.start()
datas = enqueuer.get()
for data in datas:
    # Use the inputs; training, evaluating, predicting.
    # ... stop sometime.
enqueuer.stop()

Должно быть бесконечным потоком данных.

Методы

get

Просмотреть исходный код

get()

Создаёт генератор для извлечения данных из очереди.

Пропускает данные, если они None.

Возвращаемые значения

Generator yielding tuples `(inputs, targets)`
    or `(inputs, targets, sample_weights)`.

is_running

Просмотреть исходный код

is_running()

start

Просмотреть исходный код

start(
    workers=1, max_queue_size=10
)

Запускает рабочие потоки обработчика.

stop

Просмотреть исходный код

stop(
    timeout=None
)

Останавливает работающие потоки и ожидает их завершения, если необходимо.

Должен вызываться тем же потоком, который вызвал start().