ale9806 / eiffel2

Neural Network architecture Visualization tool

Eiffel2

current version: 0.0.4
Next release: :c not soon

Neural Network architecture Visualization tool

You can use eiffel2, which you can install using pip:

python -m pip install eiffel2

to update eiffel:

python -m pip  install eiffel2 --upgrade

Just import builder from eiffel and provide a list of neurons per layer in your network as an input.

Example:

from eiffel2 import builder
builder([1, 10, 10, 5, 5, 2, 1])
# or the following if you want to have a dark theme
builder([1, 10, 10, 5, 5, 2, 1], bmode="night")

output: (normal mode)

 

 

 

output: (with bmode="night')

 

 

 

 

COLORS:

You can add colors to your network with the net_colors variable in builder() , keep in mind that for a n layer network you must select n-1 colors Eiffel2 is compatible with vanila python colors ("red","blue") and Hex colors

Online Color tool

EXAMPLE

from eiffel2 import builder
builder([1,6,6,5,2],net_colors=["#581845","#900C3F","red","#FF5733"])

output:

 

 

 

 

PYPI PAGE

eiffel2's Issues

Is there any way to save the generated visualization as jpg or pdf

Hello,

First of all, thanks for building an amazing tool to visualize the neural network architectures.

The visualization can be more fine-tuned for complex network types (such as (128, 64, 32, 16, 8, 4)).

Also, is there any way to save the generated visualization as a pdf or picture?

thank you in advance