add training set and testing set