Plot Neural Network Python