Keras输出网络结构图

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

pip uninstall pydot

pip install pydotplus

然后找到keras里面的utils\vis_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,再进入就可以了

猜你喜欢

转载自blog.csdn.net/qq_41598072/article/details/109894434
今日推荐