Skip to main content

plot model

Project description

plot_model

GitHub:https://github.com/Qinbf/plot_model.git

plot_model is a API for model visualization reference to tensorflow.keras.utils.plot_model.


Installation

Before installing plot_model, please install one of its engines: TensorFlow, Keras.

You may also consider installing the following :

pip install pydot
pip install pydot_ng
pip install graphviz
  • Finally,download and install graphviz to plot model graphs.

Then, you can install plot_model itself. There are two ways to install plot_model:

  • Install plot_model from PyPI (recommended):

Note: These installation steps assume that you are on a Linux or Mac environment. If you are on Windows, you will need to remove sudo to run the commands below.

sudo pip install plot_model

If you are using a virtualenv, you may want to avoid using sudo:

pip install plot_model
  • Alternatively: install plot_model from the GitHub source:

First, clone plot_model using git:

git clone https://github.com/Qinbf/plot_model.git

Then, cd to the plot_model folder and run the install command:

cd plot_model
sudo python setup.py install

Getting started

API is similar to tensorflow.keras.utils.plot_model

from plot_model import plot_model
plot_model(model)

default parameters:

plot_model(model, to_file='model.png', show_shapes=True, show_layer_names=False, rankdir='TB', expand_nested=False, style=0, color=True, dpi=96)

color: whether to display color. Default True.

style: 0 new style. 1 original style. Default 0.


Examples

ResNet

Inception

ResNeXt

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for plot-model, version 0.19
Filename, size File type Python version Upload date Hashes
Filename, size plot_model-0.19-py3-none-any.whl (6.8 kB) File type Wheel Python version py3 Upload date Hashes View
Filename, size plot_model-0.19.tar.gz (5.7 kB) File type Source Python version None Upload date Hashes View

Supported by

Pingdom Pingdom Monitoring Google Google Object Storage and Download Analytics Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page