A simple and yet flexible python script to build 2D Neural Networks using Graphviz library. This repository also contains a Jupyter Notebook with the script which helps in checking the correctness of the graph quickly without leaving the Jupyter Notebook workspace.
- Importing the graphviz module
- Set the Neural Network architecture defining parameters
- Create nodes
- Create edges between the created nodes
- Render and
- Save/export
The following is an example graph of a 5 layered Neural Network with 3, 4, 4, 10, 10 nodes in each layer with DENSE connections in the starting layers and SPARSE connections between the last two layers.