首页 > Keras输出网络结构图

Keras输出网络结构图

pydot已经停止开发了,python3.5以上已经用不起来了。对策是:

pip uninstall pydot

pip install pydotplus

然后找到keras里面的utilsvis_utils.py,把里面的pydot的都替换成pydotplus

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

vis_utils.py的位置可以先进入keras.utils,然后找到 from .vis_utils,再进入就可以了

更多相关: