draw neural network diagram python