Train Validation Test Split Python Syntax