Python Code For Neural Network