这个模块安装的坑比较多,所以整理如下:
请按如下顺序安装:
sudo pip install graphviz(安装接口)
sudo apt-get install graphviz(安装软件本身)sudo pip install pydot==1.1.0
sudo pip install keras
运行效果(随便举个例子,model是之前定义好的):
from keras.utils.vis_utils import plot_model
plot_model(model, to_file='model.png', show_shapes=True)
Image('model.png')