A Library for visualizing Neural Networks.
Project description
Neural Plot
Neural Plot is a python library for visualizing Neural Networks. It helps to plot Keras/Tensorflow model with matplotlib backend.
Installation
Neural Plot requires Python 3.x to run. Run the following to install:
pip install neuralplot
Example
# Importing Libraries
from neuralplot import ModelPlot
import tensorflow as tf
import numpy as np
# Uncomment while using Colab.
# %matplotlib inline
# Uncomment while using jupyter notebook. This feature is not working in colab.
%matplotlib notebook
#Creating Model
X_input = tf.keras.layers.Input(shape=(32,32,3))
X = tf.keras.layers.Conv2D(4, 3, activation='relu')(X_input)
X = tf.keras.layers.MaxPool2D(2,2)(X)
X = tf.keras.layers.Conv2D(16, 3, activation='relu')(X)
X = tf.keras.layers.MaxPool2D(2,2)(X)
X = tf.keras.layers.Conv2D(8, 3, activation='relu')(X)
X = tf.keras.layers.MaxPool2D(2,2)(X)
X = tf.keras.layers.Flatten()(X)
X = tf.keras.layers.Dense(10, activation='relu')(X)
X = tf.keras.layers.Dense(2, activation='softmax')(X)
model = tf.keras.models.Model(inputs=X_input, outputs=X)
modelplot = ModelPlot(model=model, grid=True, connection=True, linewidth=0.1)
modelplot.show()
modelplot = ModelPlot(model=model, grid=False, connection=True, linewidth=0.1)
modelplot.show()
License
MIT
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
neuralplot-0.0.2.tar.gz
(2.2 kB
view hashes)
Built Distribution
Close
Hashes for neuralplot-0.0.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 38b78422c440aab7e11756c2d8d9465f8a429e4277f2e2cf558f4d368738386b |
|
MD5 | 362c0a9fe15f67c47e19f9da9c164cc8 |
|
BLAKE2b-256 | efaf8fda2c8feb07a55ec1e7ababbbcf05280d27c0f369d6e59e8c8f619156d5 |