Handle imbalance, sampling, and overfitting | LinkedIn Interview Question