I have a dataset with high class imbalance. So to create proper training and testing sets, I have decided to use StratifiedShuffleSplit and a weighted sampler to improve training.