Neural Network Binary Classification Python Code