【发布时间】:2020-05-21 15:59:08
【问题描述】:
在 tensorflow 2.0 中,一旦模型训练好,我们如何保存模型以防止从一开始就对其进行重新训练?
【问题讨论】:
-
这已经被问及回答过,例如见stackoverflow.com/a/55770331/349130
标签: python python-3.x tensorflow keras tensorflow2.0
在 tensorflow 2.0 中,一旦模型训练好,我们如何保存模型以防止从一开始就对其进行重新训练?
【问题讨论】:
标签: python python-3.x tensorflow keras tensorflow2.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
【讨论】: