У меня есть обученная сеть 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
с выхода генератора?
Есть ли вообще лучший подход?