how to create a prediction model using python