Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[WIP] utils.visualize_util improvements #1049

Merged
merged 1 commit into from Dec 8, 2015

Conversation

julienr
Copy link
Contributor

@julienr julienr commented Nov 20, 2015

This improves visualize_util.plot with :

  • Support for container layers (using recursive=True to show container details)
  • Shape information in the plot (using show_shape=True)

Here is a notebook with some test networks showing what this does.

I haven't tested it extensively, so I would consider it work in progress for now. I would also like feedback for further improvements & options.

…rsion when plotting.

Also add an option to show layer shapes
@fchollet
Copy link
Member

Do you have some sample outputs? I'd be interested to check it out.

@julienr
Copy link
Contributor Author

julienr commented Nov 26, 2015

There is an example notebook here :
https://github.com/julienr/ipynb_playground/blob/master/keras/visutils_model2dot.ipynb
(nbviewer link if the github renderer doesn't work)

Or do you mean something else ?

@fchollet
Copy link
Member

Cool, thanks. We'll merge this after we release Keras 0.3.

fchollet added a commit that referenced this pull request Dec 8, 2015
utils.visualize_util improvements
@fchollet fchollet merged commit 3620f02 into keras-team:master Dec 8, 2015
@julienr julienr deleted the visutil_improve branch January 3, 2016 10:08
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants