Керас: Как использовать fit_generator с несколькими сетями - PullRequest
0 голосов
/ 02 февраля 2020

У меня есть обученная сеть model1 и обучаемая модель model2. Мой генератор генерирует два типа входных данных, один для обученной модели и один для неподготовленной модели. Данные были получены из того же источника, и корреляция важна.

 def generator(self,src_input):

     while 1:
         #some preperations
         yield x_for_trained_network, x_for_untrained_network,label

Моя цель - использовать генератор для получения данных, прогнозирования по первой модели и обучения по второй. Ранее я использовал:

model.fit_generator(generator = generator, steps_per_epoch=steps_per_epoch,
                    validation_data = val_generator, validation_steps = val_steps)

Для обучения model2, где мой генератор дал только 2 выхода yield x_for_untrained_network,label. Как мне интегрировать прогноз из model1? Как получить доступ к выходу генератора с помощью model.fit_generator() И как я могу сказать model2 игнорировать x_for_trained_network с выхода генератора?

Есть ли вообще лучший подход?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...