Is there a way to visualize Keras model?
I'm designing a neural network model and would like to plot the architecture with weights and shapes. How to do that?
2 Answers
JW
1There 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'))Reply
JW
1There 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()Reply
Couldn't find what you were looking for?and we will find an expert to answer.
How helpful was this page?