Open dafnevk opened 7 years ago
This indeed seems to give a visually nicer outcome (https://keras.io/visualization/) But it would add the need for a number of additional packages. Most "complicated" maybe the need for GraphViz (https://www.graphviz.org/).
Maybe for the future we can skip making a complicated plotting function here, and simply combine elements we made for the manuscript figure to match the generated architectures?
Great work, Dafne and team.
You can do that with model_to_dot in the Keras utilities as well as SVG from iPython.display. In the notebook, then use display(SVG(...)).