add splitTrainTest