Keras:找出层数
Posted
技术标签:
【中文标题】Keras:找出层数【英文标题】:Keras: find out the number of layers 【发布时间】:2018-06-24 06:10:30 【问题描述】:有没有办法获取 Keras 模型中的层数(不是参数)?
model.summary()
信息量很大,但从中获取层数并不简单。
【问题讨论】:
【参考方案1】:model.layers
会给你所有层的列表。因此这个数字是len(model.layers)
【讨论】:
len(model.layers)
无法计算“子层”,即模型中存在的层,充当您的 ubermodel 中的层。【参考方案2】:
要获得图层的图形视图,您可以使用:
from keras.utils.vis_utils import plot_model
plot_model(model, to_file='layers_plot.png', show_shapes=True, show_layer_names=True)
您需要pip install pydot
并从https://graphviz.gitlab.io/download/
下载并安装graphviz
附上一个
sample output image
【讨论】:
以上是关于Keras:找出层数的主要内容,如果未能解决你的问题,请参考以下文章