http://www.pythonreader.com/2024/07/split-your-dataset-with-scikit-learns.html