add train_set index