[ Keras ] ——基本使用:(6) 模型可视化

一、将设计模型保存成图片

参考: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 文件就能看到。

猜你喜欢

转载自blog.csdn.net/jdzwanghao/article/details/81428499
今日推荐