【问题标题】:How to load and save a model in tensorflow 2.0? [duplicate]如何在 tensorflow 2.0 中加载和保存模型? [复制]
【发布时间】:2020-05-21 15:59:08
【问题描述】:

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

【问题讨论】:

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


【解决方案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

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2020-03-16
    • 1970-01-01
    • 1970-01-01
    • 2019-09-18
    • 2019-04-12
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多