tfgraphviz

A simple graphviz wrapper to visualize a graph like TensorFlow and TensorBoard

tfgraphviz is a module to create a TensorBoard-ish graph in the DOT language of the Graphviz. tfgraphviz provides a visualization of tensorflow graph on Jupyter Notebook without TensorBoard.

Links

• GitHub: http://github.com/akimach/tfgraphviz
• PyPI: https://pypi.python.org/pypi/tfgraphviz
• Jupyter Notebook: https://nbviewer.jupyter.org/github/akimach/tfgraphviz/blob/master/examples/jupyter_sample.ipynb

Installation

Use pip to install:

$ pip install tfgraphviz

The only dependency is Graphviz.

Quickstart

import tensorflow as tf
import tfgraphviz as tfg
a = tf.constant(1, name="a")
b = tf.constant(2, name="b")
c = tf.add(a, b, name="add")
g = tfg.board(tf.get_default_graph())
g.view()

License

This package is distributed under the MIT license.

Author

Akimasa KIMURA