# 保存最好模型
checkpoint = ModelCheckpoint(filepath="./models/recommendsys_{epoch:03d}_{loss:.5f}.h5",
monitor = "loss",
verbose = 1,
save_best_only = "True",
mode = "auto",
period = 1)
callback_lists = [checkpoint]
history = model2.fit([train[0], train[1]], train[2], epochs=1000, verbose=1,callbacks=callback_lists)
两个verbose,一个控制显示是否提升保存模型,一个控制是否显示每批次损失。
如果把模型地址写成固定的就成了只保存最好模型了。