train data 和 test data各取多少