Run EDA and train models while preventing overfitting | Imc Interview Question