contrib.keras.models.save_model

tf.contrib.keras.models.save_model

tf.contrib.keras.models.save_model

save_model(
    model,
    filepath,
    overwrite=True,
    include_optimizer=True
)

Defined in tensorflow/contrib/keras/python/keras/models.py.

Save a model to a HDF5 file.

The saved model contains: - the model's configuration (topology) - the model's weights - the model's optimizer's state (if any)

Thus the saved model can be reinstantiated in the exact same state, without any of the code used for model definition or training.

Arguments:

model: Keras model instance to be saved.
filepath: String, path where to save the model.
overwrite: Whether we should overwrite any existing
    model at the target location, or instead
    ask the user with a manual prompt.
include_optimizer: If True, save optimizer's state together.

Raises:

ImportError: if h5py is not available.

© 2017 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/contrib/keras/models/save_model

在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部