tf.keras.models.load_model()加载模型

加载模型

通过 tf.keras.Sequential 的save() 方法保存的模型,可以通过下面函数加载

语法

tf.keras.models.load_model(
    filepath,
    custom_objects=None,
    compile=True
)
参数说明:
  • filepath:要加载的文件路径
  • compile:加载后是否编译模型,默认为True
    • 为True时,如果模型有 优化器,则编译。
      否则,将无法编译,并显示警告。
    • 为False时,不编译,无警告。
返回值:
  • Keras模型实例

例子

model=tf.keras.models.load_model("手写数字识别.h5")

原文出处:/show_1EF4MEf5sNnZ.html