首页 > 其他 > 详细

keras 网络结构可视化

时间:2019-11-01 16:10:53      阅读:233      评论:0      收藏:0      [点我收藏+]

1.安装:

conda install pygot
conda install graphviz

2.代码:(在test.py中)

导入包

from keras.utils.vis_utils import plot_model

在模型后面利用plot_model()函数,如:

model = models.convert_model(model)
plot_model(model, to_file=model.png, show_shapes=True, show_layer_names=True)

 

keras 网络结构可视化

原文:https://www.cnblogs.com/iamdongyang/p/11777100.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!