keras模型可视化及解决'Failed to import pydot'问题
keras.utils.vis_utils
模块提供了画出Keras模型的函数(利用graphviz)
该函数将画出模型结构图,并保存成图片:
from keras.utils import plot_model plot_model(model, to_file='model.png')
plot_model
接收两个可选参数:
show_shapes
:指定是否显示输出数据的形状,默认为False
show_layer_names
:指定是否显示层名称,默认为True
2.需要安装pydot-ng 和 graphviz依赖库
分别sudo pip install之后还是会报错 Failed to import pydot. You must install pydot and graphviz for `pydotprint` to work,可以通过 sudo apt-get install graphviz 解决.