在tensorflow 2.0中,一旦模型被训练,我们如何保存模型以防止从一开始就对其进行训练?
答案 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