加载模型
通过 tf.keras.Sequential 的save() 方法保存的模型,可以通过下面函数加载
语法
tf.keras.models.load_model(
filepath,
custom_objects=None,
compile=True
)
参数说明:
- filepath:要加载的文件路径
- compile:加载后是否编译模型,默认为True
- 为True时,如果模型有 优化器,则编译。
否则,将无法编译,并显示警告。 - 为False时,不编译,无警告。
- 为True时,如果模型有 优化器,则编译。
返回值:
- Keras模型实例
例子
model=tf.keras.models.load_model("手写数字识别.h5")