Update prediction model