Artificial Neural Network Python Code