关于Keras模型可视化教程及关键问题的解决

时间:2021-05-22

keras提供简单方便的模型可视化工具,只需一行代码就可以用框图的形式可视化出你搭建的网络结构。对于复杂网络而言,这个工具就是个神器呀。

这篇文章是解决win10环境下的keras模型可视化工具所遇到的问题,Linux暂时还没有尝试(不过也可以借鉴)。

大致环境是:

OS: win10 python==3.5.4 keras==2.2.0 pydot==1.2.4 pydot_ng==1.0.0

首先,你可以查看keras官方文档教程。

from keras.utils import plot_modelplot_model(model, to_file='model.png')

这两行代码实现可视化?

其实还是有些问题的,因为你并没有安装GraphViz。

教程

pip install graphvizpip install pydotpip install pydot_ng

注意:光用pip安装以上两个模块还是不够的

还需要安装GraphViz:

1. http://pile(loss='binary_crossentropy', optimizer=sgd, metrics=['accuracy']) # 神经网络可视化 plot_model(model, to_file='model.png') if __name__ == '__main__': run()

运行结果如下:

顺便跑了以下VAE的网络结构:

整体上,效果不错。主要是,使用方便。

以上这篇关于Keras模型可视化教程及关键问题的解决就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章