keras 显示模型结构

  1. model.summary(line_length=150, positions=[0.30, 0.60, 0.7, 1.]
    在这里插入图片描述

  2. plot_model(model, to_file=’./model.png’, show_shapes=True)
    需要 from keras.utils import plot_model
    基于以下:
    安装pydotplus库 将出错文件内pydot(python3.6不支持了)替换为 pydotplus
    pip install pydotplus
    安装graphviz库
    sudo apt-get install graphviz
    在这里插入图片描述

发布了8 篇原创文章 · 获赞 0 · 访问量 3353

猜你喜欢

转载自blog.csdn.net/APAN1013/article/details/101456987
今日推荐