CODEWITHSUNDEEP
pythontensorflowkerasdeep-learning
Reda Abdellah KAMRAOUI
Reda Abdellah KAMRAOUI

Reputation: 51

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

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.

enter image description here

Upvotes: 5

Views: 3385

Answers (1)

Andrew Louw
Andrew Louw

Reputation: 687

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.

Upvotes: 4

Related Questions