Skip to main content

Generate a neural network architecture Image

Project description

Neural Network Visualizer

General Description

A module which creates a neural network with the given architecture(only Dense layers)

Installation

Before installation

Before installing the module, run the below command at your prompt to install the graphviz

$ sudo apt install graphviz

Normal installation

$ sudo pip3 install neuralnet-visualize

Development installation

$ git clone https://github.com/AnuragAnalog/nn_visualize.git
$ cd nn_visualize

After installation

After installing the module, if you want to upgrade the module, run the below command.

sudo pip3 install neuralnet-visualize --upgrade

Future Works

  • Add Convolutional layers, Maxpooling, Flatten layers, LSTM's
  • Specific colors for activation functions
  • Specific colors for types of layers

Neural Network Visualizer Version History

0.1.0

  • Added documentation

0.0.4

  • Refactored summarize function
  • Added a code part which colors output layer to red
  • Added from_tensorflow method

0.0.3

  • Added more display orientations
  • Refactored the build network and add layer code
  • Added some more file extensions to which image can be saved
  • Added summarize function

0.0.2

  • Added colors to type of layers
  • Added Dense layers

0.0.1

Intial Version

  • Starting code

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

neuralnet_visualize-0.1.0.tar.gz (4.0 kB view hashes)

Uploaded Source

Built Distribution

neuralnet_visualize-0.1.0-py3-none-any.whl (8.5 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page