Keras model.summary()的输出很好吗?

时间:2019-05-29 12:34:25

标签: python tensorflow keras deep-learning

有可能有一个很好的keras model.summary()输出,它可以包含在纸中,也可以绘制在一个像这样的漂亮表中。

enter image description here

1 个答案:

答案 0 :(得分:2)

您需要安装graphvis和pydot,但您可能希望由此获得结果。它没有表格,但我认为图形要好得多。

    from keras.utils import plot_model
    plot_model(model, to_file='model.png', show_shapes=True,show_layer_names=True)

但是,如果要将多个图层嵌套在一起,则必须制作正确命名的子模型。