Remove training set after training.