Add predictProbability fro BaseModel