模型可以保存在一个文件中,其中包含权重值,模型的配置以及优化器的配置。
另外在keras中,我们可以使用tensorflow.js来加载模型,这样可以模型在浏览器中运行
keras保存模型采用HDF5格式
# 保存模型model.save('first_model.h5')# 使用保存的模型new_model = tf.keras.models.load_model('first_model.h5')# 仅保存模型架构json_config = model.to_json()# 重建模型用上面的json数据rebuild_model = tf.keras.models.model_from_json(json_config)# 仅保存模型的权重weights = model.get_weights()# 设置权重model.set_weights(weights)
在训练期间怎么去保存检查点能?
在模型训练的过程中,采用回调函数的方式来保存我们在训练过程中产生的参数
cp_callback = tf.keras.callbacks.ModelCheckpoint('cp.ckpt')# 然后在训练阶段添加回调函数#
