visualize_util -> vis_utils

satels · Feb 28, 2017 · 0e9ac3d · 0e9ac3d
1 parent 4e50446
commit 0e9ac3d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/docs/templates/visualization.md b/docs/templates/visualization.md
@@ -2,15 +2,15 @@
 ## Model visualization
 
 The `keras.utils.visualize_util` module provides utility functions to plot
-a Keras model (using graphviz).
+a Keras model (using `graphviz`).
 
 This will plot a graph of the model and save it to a file:
 ```python
-from keras.utils.visualize_util import plot
-plot(model, to_file='model.png')
+from keras.utils import plot_model
+plot_model(model, to_file='model.png')
 ```
 
-`plot` takes two optional arguments:
+`plot_model` takes two optional arguments:
 
 - `show_shapes` (defaults to False) controls whether output shapes are shown in the graph.
 - `show_layer_names` (defaults to True) controls whether layer names are shown in the graph.

diff --git a/keras/utils/visualize_util.py → keras/utils/vis_utils.py b/keras/utils/visualize_util.py → keras/utils/vis_utils.py
@@ -86,7 +86,10 @@ def model_to_dot(model, show_shapes=False, show_layer_names=True):
     return dot
 
 
-def plot(model, to_file='model.png', show_shapes=False, show_layer_names=True):
+def plot_model(model,
+               to_file='model.png',
+               show_shapes=False,
+               show_layer_names=True):
     dot = model_to_dot(model, show_shapes, show_layer_names)
     _, extension = os.path.splitext(to_file)
     if not extension: