improve doc in predict