is there a nice output of Keras model.summary( )?

我的未来我决定 提交于 2021-02-08 04:51:32

问题


is it possible to have a nice output of keras model.summary(), that can be included in paper, or can be ploted in a nice table like this.


回答1:


You need to install graphvis and pydot, but you might like the results from this. It doesn't make a table but the graph is much better in my opinion.

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

But you would have to make properly named sub models if you want to nest the several layers together.



来源:https://stackoverflow.com/questions/56360776/is-there-a-nice-output-of-keras-model-summary

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!