Keras中模型的可视化,import pydot的一些报错解决

在Keras 中文文档中可以看到,使用keras.utils.vis_utils模块绘制Keras模型,
具体实现代码如下:

from keras.utils import plot_model
plot_model(model, to_file='model.png') #导出model模型图到程序所在路径下的model.png中

plot_model 有 4 个可选参数:

show_shapes (默认为 False) 控制是否在图中输出各层的尺寸。
show_layer_names (默认为 True) 控制是否在图中显示每一层的名字。
expand_dim(默认为 False)控制是否将嵌套模型扩展为图形中的聚类。
dpi(默认为 96)控制图像 dpi。

调用pydot的报错解决方法

环境: windows10、Python3.6、Keras2.3.0

1. 安装pydot

由于Keras中使用了pydot.find_graphviz()函数,而这个函数在1.2.*之后的版本中被剥离了,所以不能直接安装pydot,而是安装pydot_ng。
1.1 安装pydot_ng

pip install pydot_ng

1.2 报错处理
在运行时出现NameError: name ‘pydot’ is not defined的错误,点击"C:\Python安装路径\Python\Python36\lib\site-packages\keras\utils\vis_utils.py"或者直接在Python安装路径打开vis_utils.py文件


                
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值