neural network code in python