https://camera-optical-zoom.blogspot.com/2023/04/splitting-data-to-training-testing-and.html