https://remp3indir.net/traintestsplit-method-machine-learning-tutorial-using-python/