Neural network visualisation

[DrJonnyT]

!pip install visualkeras
import visualkeras
visualkeras.layered_view(model,legend=True,scale_xy=1,scale_z=1, max_z=100)

Makes a nice plot, compared to just model.summary()