A library for generating machine learning figures and illustrations

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.

AI-FIG

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

GitHub

The source code and more information is available on GitHub:

github.com/evgiz/aifig

And the project pypi page:

pypi.org/project/aifig/