https://tsoli.github.io/posts/splitting-data-for-machine-learning/