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.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.0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 8f7c347118db33f0e435ba326d6f1b698729df099b2a171a638f041bb42abc91 |
|
MD5 | 8d1dcd6cd9509c37c2158c07c9845620 |
|
BLAKE2b-256 | ed2ac0fee39fd8ba5a54adfbbdeb7e85af89fea153de036739daa24b40912f78 |
Close
Hashes for neuralnet_visualize-0.2.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 96c5d3f570c528c302ba1e5336344e898926840f561772a16837f1d6d0c725cc |
|
MD5 | aa4063710124501a2295fcf8eeb6a67d |
|
BLAKE2b-256 | f94a93f4727f701651bf44e303a77a5f2fc88fa7ff2b1df225ebf5f6bc3c277e |