add the trainging set and testing set