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模型可视化教程及关键问题的解决就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。
最新文章
热门文章
猜你喜欢
