如何在Tensorflow 2.0中加载和保存模型?

时间:2020-02-05 17:10:57

标签: python python-3.x tensorflow keras tensorflow2.0

在tensorflow 2.0中,一旦模型被训练,我们如何保存模型以防止从一开始就对其进行训练?

1 个答案:

答案 0 :(得分:0)

训练完模型后,要保存,可以使用以下代码:

model.save('model.h5')

model是我的模型的名称,“ model.h5”将是包含经过训练的模型的文件的名称,其中“ .h5”是扩展名。

H5文件是以分层数据格式(HDF)保存的数据文件。它包含科学数据的多维数组。

现在,要加载此保存的模型,您可以使用以下代码:

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


请看以下示例,以更好地理解以下内容:
https://github.com/TheArhaam/Python-ML-Basics/blob/master/demo6.py