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

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

Методы

get

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

get()

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

Пропускает данные, если они None. Возвращает: генератор, возвращающий кортежи (inputs, targets) или (inputs, targets, sample_weights).

is_running

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

is_running()

start

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

start(
    workers=1, max_queue_size=10
)

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

stop

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

stop(
    timeout=None
)

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

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