Plotting an Architecture of Tensorflow Keras Model by plot_model()

This post shows how to plot an architecture flow of Tensorflow Keras model by using plot_model() function.

To plot the architecture flow of Tensorflow Keras model by using plot_model() function, two packages should be installed: graphviz and pydot.

To install these packages, type the following commands in the anaconda prompt.

pip install graphviz
pip install pydot

In the Jupyter notebook, we can use plot_model() function to draw an architecture of the following Keras model.

from keras import layers, models
from keras.utils.vis_utils import plot_model
#1. Modeling
model = models.Sequential()
model.add(layers.Dense(128, activation='relu', 
                       input_shape=(256,),       name='HiddenLayer01'))
model.add(layers.Dense(64 , activation='relu',   name='HiddenLayer02'))
model.add(layers.Dense(32 , activation='relu',   name='HiddenLayer03'))
model.add(layers.Dense(1  , activation='softmax',name='OuputLayer'))
              optimizer = 'adam', metrics=['accuracy'])
#2. Visualizing
plot_model(model, show_shapes=True, to_file='model.png')

As expected, we can get the following visualization.


If you happen to encounter an error regarding the installation, one possible solution is to use the following conda command in the anaconda prompt.

conda install graphviz

Visit SH Fintech Modeling for additional insights on this topic:

Disclosure: Interactive Brokers

Information posted on IBKR Campus that is provided by third-parties does NOT constitute a recommendation that you should contract for the services of that third party. Third-party participants who contribute to IBKR Campus are independent of Interactive Brokers and Interactive Brokers does not make any representations or warranties concerning the services offered, their past or future performance, or the accuracy of the information provided by the third party. Past performance is no guarantee of future results.

This material is from SHLee AI Financial Model and is being posted with its permission. The views expressed in this material are solely those of the author and/or SHLee AI Financial Model and Interactive Brokers is not endorsing or recommending any investment or trading discussed in the material. This material is not and should not be construed as an offer to buy or sell any security. It should not be construed as research or investment advice or a recommendation to buy, sell or hold any security or commodity. This material does not and is not intended to take into account the particular financial conditions, investment objectives or requirements of individual customers. Before acting on this material, you should consider whether it is suitable for your particular circumstances and, as necessary, seek professional advice.