Small repository to plot neural network visualisation from keras model.
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
plotnn
.gitignore
LICENSE
README.md
setup.cfg
setup.py

README.md

Plotnn

Description

Plot Neural Networks is a visualization python library inspired by ANN Visualizer. This library is working with Keras model interface. It uses python's graphviz library to create a presentable graph of the neural network you are building. Working for sequential and non-sequential models.

For example if you have this keras model as input:

example keras model

You will have this visualisation:

example visualisation

Install

pip install -e plotnn/

Usage

from plotnn.visualize import model_visualize
#Build your model here
model_visualize(model, name="graph")