Neural Network Python Example For Loop