在使用tensorflow2.0 keras时, 可使用keras.utils.plot_model
方法方便的画出模型的拓扑结构图。 这个方法依赖于pydot
和graphviz
, 若不安装, 执行到这个方法时,会报错:
File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/keras/utils/vis_utils.py", line 283, in plot_model
dpi=dpi)
File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/keras/utils/vis_utils.py", line 110, in model_to_dot
raise ImportError('Failed to import pydot. You must install pydot'
ImportError: Failed to import pydot. You must install pydot and graphviz for `pydotprint` to work.
安装方式为:
sudo apt install graphviz
pip install pydot --user
需要注意的是,pip中也有graphviz
, 但安装后不管用, 一定要用apt安装。