keras模型可视化pydot_ng和graphviz在python3.6环境下的安装和配置(Windows10)

最新推荐文章于 2023-08-30 08:11:20 发布

原创

最新推荐文章于 2023-08-30 08:11:20 发布 · 1.5k 阅读

18 ·

CC 4.0 BY-SA版权

在Windows10上使用Keras的plot_model功能时遇到错误，通过以下步骤解决：1. 使用Anaconda安装Graphviz；2. 下载并安装Graphviz软件，将bin目录添加到PATH环境变量；3. 由于conda无法安装pydot-ng，采用pip安装；4. 修改pydot-ng和keras的vis_utils.py文件，确保引入正确的库和配置Graphviz路径。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

使用keras的函数plot_model可视化模型

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

运行结果出错

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-40-762315582316> in <module>
     43 
     44 # Plot model graph
---> 45 plot_model(model, show_shapes=True, show_layer_names=True, to_file='model.png')
     46 Image(retina=True, filename='model.png')

E:\Software\Anaconda3\envs\tensorkeras\lib\site-packages\keras\utils\vis_utils.py in plot_model(model, to_file, show_shapes, show_layer_names, rankdir)
    130             'LR' creates a horizontal plot.
    131     """
--> 132     dot = model_to_dot(model, show_shapes, show_layer_names, rankdir)
    133     _, extension = os.path.splitext(to_file)
    134     if not extension:

E:\Software\Anaconda3\envs\tensorkeras\lib\site-packages\keras\utils\vis_utils.py in model_to_dot(model, show_shapes, show_layer_names, rankdir)
     53     from ..models import Sequential
     54 
---> 55     _check_pydot()
     56     dot = pydot.Dot()
     57     dot.set('rankdir', rankdir)

E:\Software\Anaconda3\envs\tensorkeras\lib\site-packages\keras\utils\vis_utils.py in _check_pydot()
     18     if pydot is None:
     19         raise ImportError(
---> 20