Visualize neural network model in python. .