Train with imbalanced sampled data | LinkedIn Interview Question