Parameter “stratify” from method “train_test_split” (scikit Learn)

Question

I am trying to use train_test_split from package scikit Learn, but I am having trouble with parameter stratify. Hereafter is the code:

Answer 1

In this context, stratification means that the train_test_split method returns training and test subsets that have the same proportions of class labels as the input dataset.