deep neural network python code