Keras output section of graph is very complex

waleedka / hiddenlayer

Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.

MIT License

1.79k stars 266 forks source link

Keras output section of graph is very complex #55

Open LanceNorskog opened 4 years ago

LanceNorskog commented 4 years ago

The output section of every Keras graph is complex, and has a long chain of TF nodes that uses up a lot of screen space.

Request some way to Prune everything downstream of the "beginning of the outputs", and rename the end block "OUTPUT". Graphs with multiple outputs might have "OUTPUT #1" etc.