Prediction Model In Python