Write a python code to draw neural network architecture