Traintestvalidation Function Python Example