Traintestvalidation Function Python Definition