Linux 拨号vps windows公众号手机端

tensorflow模型怎么保存与加载

lewis 5年前 (2020-02-10) 阅读数 13 #大数据
文章标签 tensorflow

在TensorFlow中,可以使用tf.keras.models.save_model()方法保存模型,该方法将模型保存为HDF5文件格式。可以使用以下代码保存模型:

model.save('my_model.h5')

加载保存的模型可以使用tf.keras.models.load_model()方法:

model = tf.keras.models.load_model('my_model.h5')

另外,也可以使用tf.saved_model.save()方法保存模型为SavedModel格式,可用于在TensorFlow Serving等地方部署模型。以下是保存模型的示例代码:

tf.saved_model.save(model, 'saved_model')

加载SavedModel格式的模型可以使用tf.saved_model.load()方法:

loaded_model = tf.saved_model.load('saved_model')
版权声明

本文仅代表作者观点,不代表米安网络立场。

发表评论:

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

热门