Neural Network Visualizer

What is NN_Visualize?

A module which creates a neural network image with the given architecture. It’s a handy tool to see how your network is built as compared to a model summary

Installation

For installation you can follow the instructions here

After the Development installation(linux), run the below command to install all the requirements

$ sudo pip3 install -r requirements.txt

Examples

Some of the examples can be found here

The docstring style which I have followed is numpydoc

Network Visualization

Future Works

• Add Convolutional layers, Maxpooling, Flatten layers
• Directly convert from pytorch models
• Add Sequence model layers
• Directly from the pickle files
• Specific colors for activation functions