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.2.1
- Added title to the image
- Optimized some code
0.2.0
- Made it independent of tensorflow
- Fixed some bugs
0.1.4
- Fixed some bugs
- Added more layers maxpooling, avgpooling, flatten
- changed the shape of conv2d layer
0.1.3
- Added class docstrings with examples
- Added some more parameters for conv layer
- Disabled custom addition of layers after from_tensorflow call
0.1.2
- Added Conv layer, with kernel size parameter
- Refactored add_layer function
- Added Conv layer to from_tensorflow
0.1.1
- Added some possible exceptions
- Restructred basic functions in while initial import
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
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for neuralnet_visualize-0.2.1.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1dbe0d2c3c2361913abbc64185bfed5e9c7d2147fab8f6948fbaab4375b21c24 |
|
MD5 | a2a21afb6b26145024b25fcc0164b4d3 |
|
BLAKE2b-256 | 09abc7acd245161a63433827af601d9d85de4b160fa2029529a017f5aa50eca8 |
Close
Hashes for neuralnet_visualize-0.2.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e0c34fa08f1bd32d2bceea9a5a5e3155f4d695eb54fe42555e7fa4319f1bc8e4 |
|
MD5 | 81db9d467fbdfc1b25b5d45da0299ee2 |
|
BLAKE2b-256 | 7c7f21bc4fdc9002584e96cce8e14c5d18838cd23ba7da2aefcf9cb05566faae |