Skip to content

This project is a example to how visualize an Artificial Neural Network architecture with keras, ANNvisualizer and graphviz

Notifications You must be signed in to change notification settings

jgabriellima/ann_visualizer

Repository files navigation

Visualizing Artificial Neural Networks (ANNs)

with just One Line of Code

ANN Visualizer is a python library that enables us to visualize an Artificial Neural Network using just a single line of code. It is used to work with Keras and makes use of python’s graphviz library to create a neat and presentable graph of the neural network you’re building.

Installation

We will need 3 libraries for this demo

  • keras
  • ANNvisualizer
  • graphviz

You can install the library using the below commands:

$ pip3 install keras
$ pip3 install ann_visualizer
$ pip install graphviz
$ pip install h5py

Running train

$ python3 train.py

Visualizing the Artificial Neural Network Model

$ python3 visualizr_model.py

About

This project is a example to how visualize an Artificial Neural Network architecture with keras, ANNvisualizer and graphviz

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages