“微信公众号”
Keras提供了模型可视化的函数,但是要依赖一些工具包,安装起来稍微麻烦了一点,还可能有坑,本篇文章记录我在安装使用过程中的步骤。
环境:win10、Anconda3-5.1.0、Python3.6、Keras2.1.5
(1)安装附加依赖项
1. pydot
pip install pydot_ng
官方文档中说直接安装pydot,但是由于Keras中使用了pydot.find_graphviz()函数,而这个函数在1.2.*之后的版本中被剥离了,所以不能直接安装pydot,而是安装pydot_ng。
在Keras的 keras.utils.vis_utils 中,import pydot或者pydot_ng的源码如下: