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.close()

enqueuer.get() должен быть бесконечным потоком данных.

Методы

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().