waleedka / hiddenlayer

Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.
MIT License
1.79k stars 266 forks source link

Ability to visualize subgraphs #56

Open LanceNorskog opened 4 years ago

LanceNorskog commented 4 years ago

I use a few libraries that have custom Keras layers, and generate complex graphs inside these layers. Some are algorithmic and generate, say, 20 nodes N times.

Some way to group the sections of nodes would be really helpful in creating visualizations. TF graph nodes have names and attribute lists.