我需要在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个模型-生成器和鉴别器。没有错误消息,但在结束训练后不保存任何模型。