如何在 tensorflow 2.0 中加载和保存模型？

Question

tensorflow 2.0中，模型训练好后，如何保存模型，防止从头开始重新训练？

Answer 1

训练完模型后，要保存，您可以使用以下代码：

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