To better visualize a machine learning model, it is useful to create diagrams and figures of your models. I have published a python library called AIFIG which is intended for this purpose.
The library allows you to easily define a graph and generate figures such as this:
Figure: Generative adversarial network model, figure generated using AIFIG
The library can be installed using pip:
pip install aifig
The source code and more information is available on GitHub:
And the project pypi page: