Predict Method In Python