add pred_raw/pred_top