GAN中的ModelCheckpoint:如何在最高F1分数上保存鉴别器模型

时间:2019-09-27 08:19:53

标签: python generative-adversarial-network gan

我需要在GAN中定义ModelCheckpoint的帮助,因此它将鉴别器模型保存在最高的F1_score上。

尝试对ModelCheckpoint使用“常规”设置,但由于网络具有两个模型(区分器和生成器),因此无法正常工作

mcp_save = ModelCheckpoint('models / usage_gan_ba128_mcp.h5',save_best_only = True,monitor ='f1',mode ='max')

f1_p =火车(X_res,y_res,              X_test,y_test,              发电机,鉴别器,              结合起来              num_classes = 2              历元= 10000,              batch_size = 128,              回调= [mcp_save])

保存了预期的2个模型-生成器和鉴别器。没有错误消息,但在结束训练后不保存任何模型。

0 个答案:

没有答案