Is there a way to visualize Keras model?

Question

I'm designing a neural network model and would like to plot the architecture. How to do that?

Saturday, March 9, 2019 4:20 PM

1 votesNNNeural Networks177.00

2 Answers

Couldn't find what you were looking for?and we will find an expert to answer.

How helpful was this page?

John Wilson · Answer 1

There is also another option

from IPython.display import SVG
from keras.utils import model_to_dot

SVG(model_to_dot(model).create(prog='dot', format='svg'))

John Wilson · Answer 2

There are 2 ways to do that. 1 plots on the terminal, the second plots into the file.

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

model.summary()