remove train_sample