Skip to content

Instantly share code, notes, and snippets.

@totetmatt

totetmatt/dot.keras.py

Created March 24, 2018 10:15
Show Gist options
  • Save totetmatt/dcc85d27b0fdfd79513cbe43201f507f to your computer and use it in GitHub Desktop.
Save totetmatt/dcc85d27b0fdfd79513cbe43201f507f to your computer and use it in GitHub Desktop.
Download ZIP
Dot export with Keras
Raw
dot.keras.py
from keras.applications import *
from keras.utils import plot_model
# [..]
# model = ...
# Get your own model here
# [..]
model = NASNetMobile() #Example with NASNetMobile
plot_model(model,show_shapes=False, to_file='model.dot')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment