add training set and training script