}
@Override
protected Sample createSample(Dataset dataset, boolean trainSample) {
RankingSample sample = new RankingSample((RankingDataset) dataset);
RankingTrainingConfig config = (RankingTrainingConfig) trainingConfig;
if (trainSample) {
if (config.augmentationDocSamplingEnabled) {
return sample.getAugmentedSampleWithDocSampling(config.augmentationDocSamplingTimes, config.augmentationDocSamplingRate, rnd);
}
}