Prediction: remove weight parameter for the while