Kalman Filter Python Neural Network