Использование входных конвейеров с tf.data
Генерация набора данных с использованием tf.data.Dataset.from_generator ()
Вывод, который я получил после подгонки модели с помощью train_dataset
Я думаю, что вы можете следовать этому примеру здесь и этому примеру . Эти два простых примера демонстрируют, как использовать конвейер входных данных. Надеюсь это поможет. Спасибо!