一、将设计模型保存成图片。
参考:https://www.jianshu.com/p/275aad639b5e
model = Model(input=inputs_res,output=outputs)
from keras.utils.vis_utils import plot_model # vis_utils和参考文章有改动。
plot_model(model, to_file='model1.png',show_shapes=True)
需要一些必要的安装包:
pip install graphviz
pip install pydot
二、超参数的查看。
json_string = model.to_json()
open( 'model_struct.json', 'w').write(json_string) # 保存模型结构
直接打开 model_struct.json 文件就能看到。