在建立神经网络模型的时候,可以调用keras中的plot_model模块画出模型示意图,方便调整模型结构:
from keras.utils.vis_utils import plot_model
plot_model(model, to_file='model2.png',show_shapes=True)
然鹅!发现事情并不简单,报错了!
于是想着照着上面的要求,安装个 pydot和graphviz,装好了还是不行!!
网上的好多解决方法都是坑,经过我苦苦搜索,终于找到一个解决方案(参考这个方法),如下:
- pip install graphviz
- 下载并安装graphviz的msi或者zip文件并安装,地址:https://graphviz.gitlab.io/_pages/Download/Download_windows.html
- 最好再配置一下环境变量(在PATH中添加路径):
- pip install pydot-ng
- 重启你的Python IDE(我用的Pycharm)即可
参考:https://blog.csdn.net/wushaowu2014/article/details/88958400