the training dataset