Prediction model for  WLE dataset