Несколько проходов на выход с помощью fit_generator ()?

Question

У меня есть приложение, в котором сбор данных из файла стоит дорого, поэтому я хочу взять пакет обучающих данных и потренироваться на нем несколько итераций, прежде чем собирать больше. Если я создаю генератор для извлечения данных из файла, похоже, что fit_generator() будет использовать эти данные только один раз, а затем снова вызовет генератор. Есть ли способ заставить fit_generator() использовать одни и те же данные для нескольких проходов обучения? Или правильное решение для встраивания этой функции в мой генератор?

Answer 1

Просто создайте новый генератор, который обернет другой генератор.

def repeater_generator(generator, reps=10):
    while True:
        batch = next(generator)
        for i in range(reps):
            yield batch

Должно быть что-то подобное.