To spit data into a training set and test set, you had indeed used the train_test_split library from scikit learn.Here we will talk about one parameter called stratify in train_test_split in a simple way.