Fix

* Add a image in README.rst
* Add keyword arguments in `tfgraphviz.board(...)`

README.rst

@@ -36,6 +36,9 @@ Quickstart
     g = tfg.board(tf.get_default_graph())
     g.view()
 
+.. image:: https://raw.githubusercontent.com/akimach/tfgraphviz/master/img/graph.jpg
+    :align: center
+
 License
 -------
 

examples/jupyter_sample.ipynb

@@ -871,7 +871,7 @@
     }
    ],
    "source": [
-    "tfg.board(calc_reg, depth=2)"
+    "tfg.board(reg_g, depth=2)"
    ]
   }
  ],

setup.py

@@ -13,7 +13,7 @@
 
 setup(
     name  = 'tfgraphviz',
-    version = '0.0.2',
+    version = '0.0.3',
     description = 'A simple graphviz wrapper to visualize a graph like TensorBoard',
     long_description = long_description,
     license = 'MIT',
@@ -22,7 +22,7 @@
     url = 'https://github.com/akimach/tfgraphviz',
     keywords = 'tensorflow tensor machine learning graphviz',
     packages = find_packages(),
-    install_requires = ['tensorflow', 'graphviz'],
+    install_requires = ['graphviz'],
     classifiers = [
       'Programming Language :: Python :: 2.7',
       'Programming Language :: Python :: 3.3',
@@ -31,4 +31,4 @@
       'Intended Audience :: Education',
       'Topic :: Scientific/Engineering :: Mathematics',
     ]
-)
+)

tfgraphviz/__init__.py

@@ -1,8 +1,6 @@
-#!/usr/bin/env python
 # coding: utf-8
-
 from .graphviz_wrapper import board
 
-__author__  = 'akimacho'
-__version__ = '0.0.1'
-__license__ = 'MIT'
+
+__version__ = '0.0.3'
+__license__ = 'MIT'

tfgraphviz/graphviz_wrapper.py

@@ -137,11 +137,11 @@ def add_edges(node_inpt_table, g):
     return g
 
 
-def board(tfgraph, depth=1, style=True):
+def board(tfgraph, depth=1, name='G', style=True):
     global CLUSTER_INDEX
     CLUSTER_INDEX = 0
     _node_table = node_table(tfgraph, depth=depth)
     _node_inpt_table = node_input_table(tfgraph, depth=depth)
-    g = add_nodes(_node_table, name='G', style=style)
+    g = add_nodes(_node_table, name=name, style=style)
     g = add_edges(_node_inpt_table, g)
     return g