Skip to content

Instantly share code, notes, and snippets.

@Thanatoz-1

Thanatoz-1/create_model_svg.py

Created September 1, 2018 21:21
Show Gist options
  • Save Thanatoz-1/d8a9331caf2ca711e1e707e0935b63e2 to your computer and use it in GitHub Desktop.
Save Thanatoz-1/d8a9331caf2ca711e1e707e0935b63e2 to your computer and use it in GitHub Desktop.
Download ZIP
Keras creating model SVG
Raw
create_model_svg.py
from IPython.display import SVG
from keras.utils.vis_utils import model_to_dot
from keras.utils import plot_model
from kt_utils import *
plot_model(happyModel, to_file='HappyModel.png')
SVG(model_to_dot(model).create(prog='dot', format='svg'))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment