Here's how I went about it.
Add the parameters that define the layers and node images to the parameter dictionary.
Now we extend our collection of parameters to cover the neural network--its layers and the individual node images. Feel free to use my solution to the last exercise as a starting point: nn_viz_03.py
Ready? Go for it.
2 comments