Linkedin Python Test Train Split