Difference Between Train Test Split In Python Example