Tensorflow Keras模型储存
Tensorflow有某种模型存储方式。第一种是存成checkpoint档(.ckpt),使用时机是训练过程中欲保存当前会话状态。第二种是存成pb档(.pb),如果模型架构已确定第三种是Keras(目前已合并到TF2.0)的save()存成HDF5档(.h5),HDF是设计本篇文章透过波士顿房价预测资料集,训练一个DNN模型并示范如何汇出与载入.pb和.h5模型档。
改造模型
首先我们使用Tensorflow Keras API建立一个顺序模型,共有两个