結果 : split data into train validation and test in python