How to Visualize Neural Network Architectures in Python