Basic Neural Network Python Code Generator