Provide keras.js visualization of network as a jupyter widget

[dattanchu]

The visualization of the network is much better than the current Keras model.summary(). Can we make the visualization component into a widget that can be used to visualize Keras model within jupyter notebook?

[brunoalano]

I am free to do that if allowed

[yassineAlouini]

That would be awesome!

[transcranial]

@dattanchu @brunoalano Excellent idea. At this point, the layout process is still somewhat manual, but there's no reason it can't be made entirely automatic.

Initially I had tried using d3 but wasn't happy with the results I was getting. So I ended up just creating a custom architecture definition file, generated from the keras model config file, with arrays for the layers and the connections.

From this, layers are created programmatically: example. The connections are SVG paths calculated from those elements: example.

[brunoalano]

I think would be easier to write a D3 Plugin that loads from the Keras JSON Model export, and then integrate into this project.

[transcranial]

I agree. I'd be more than happy to support that.