代码之家  ›  专栏  ›  技术社区  ›  Simbarashe Timothy Motsi clayton.nichols

如何使用keras将检查点保存到路径?

  •  0
  • Simbarashe Timothy Motsi clayton.nichols  · 技术社区  · 7 年前

    如何在keras中保存一个检查点并将其用于回调,该文件将需要一个.h5模型。

    1 回复  |  直到 7 年前
        1
  •  1
  •   today    7 年前

    您需要设置.h5模型的路径目录,并定义检查点的参数。
    我的建议是 monitor= val_acc 并将save_best_设置为true。

    filepath="model.h5"
    checkpoint1 = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=True, mode='max')
    save_best_only=True, mode='min')
    callbacks_list = [checkpoint1]