Draw Neural Network Diagram Python