ModelCheckpoint в GAN: как сохранить модель дискриминатора с наибольшим результатом F1 - PullRequest
0 голосов
/ 27 сентября 2019

Мне нужна помощь в определении ModelCheckpoint в GAN, поэтому он сохраняет модель дискриминатора на самом высоком F1_score.

Пробовал использовать "обычную" настройку для ModelCheckpoint, но он не работает - правильно, потому что в сети есть две модели (дискриминатор и генератор)

mcp_save = ModelCheckpoint ('models / creation_gan_ba128_mcp.h5'), save_best_only = True, монитор = 'f1', режим = 'max')

f1_p = train (X_res, y_res, X_test, y_test, генератор, дискриминатор, комбинированный, num_classes = 2, epochs = 10000, batch_size)= 128, обратные вызовы = [mcp_save])

Ожидается сохранение 2 моделей - Генератор и дискриминатор.Нет сообщения об ошибке, но не сохраняет модель по окончании тренировки.

...