Handle imbalance, validate samples, and avoid overfitting | LinkedIn Interview Question