GitHunt
VL

vlasenkoalexey/tfgraphviz

A visualization tool to show a TensorFlow's graph like TensorBoard

tfgraphviz

.. image:: https://img.shields.io/github/license/akimach/tfgraphviz.svg
:alt: GitHub license
:target: https://github.com/akimach/tfgraphviz/blob/master/LICENSE

.. image:: https://badge.fury.io/py/tfgraphviz.svg
:target: https://badge.fury.io/py/tfgraphviz

tfgraphviz is a module to visualize a TensorFlow's data flow graph like TensorBoard using Graphviz. tfgraphviz enables to provide a visualization of tensorflow graph on Jupyter Notebook without TensorBoard.

Installation

Use pip to install:

.. code-block:: bash

$ pip install graphviz
$ pip install tfgraphviz

The only dependency is Graphviz.

macOS:

.. code-block:: bash

$ brew install graphviz

Ubuntu:

.. code-block:: bash

$ apt-get install graphviz

Quickstart

.. code-block:: python

import tensorflow as tf
import tfgraphviz as tfg

g = tf.Graph()
with g.as_default():
    a = tf.constant(1, name="a")
    b = tf.constant(2, name="b")
    c = tf.add(a, b, name="add")
tfg.board(g)

.. image:: https://raw.githubusercontent.com/akimach/tfgraphviz/master/img/graph.jpg
:align: center

License

This package is distributed under the MIT license <https://raw.githubusercontent.com/akimach/tfgraphviz/master/LICENSE>_.

Author

Akimasa KIMURA <https://github.com/akimach>_

Languages

Python100.0%

Contributors

MIT License
Created December 8, 2020
Updated April 24, 2024
vlasenkoalexey/tfgraphviz | GitHunt