There are several ways to split the data set into training data set and test data set.
In this blog, I will talk about the difference between these approaches.
sklearn.model_selection.train_test_split
Doc: https://scikit-learn.org/stable/modules/generated/sklearn.model_selection.train_test_split.html
sklearn.model_selection.ShuffleSplit
Doc: https://scikit-learn.org/stable/modules/generated/sklearn.model_selection.ShuffleSplit.html