Methods

The good part about stratification is that there it is easy and in fact, there are a lot of libraries to do so.

1. library - Sklearn

sklearn.model_selection_train_test_split( stratify = data["variable"] )