@Override
protected LearningModule getLearningModule(String name) throws Exception {
int maxTrainInstances = getMaxTrainInstances();
if (name.equals("LambdaMART")) {
LambdaMART learner = new LambdaMART();
learner.init(configHolder, (RankingDataset) trainDataset, maxTrainInstances, (validDataset != null ? validDataset.numInstances
: trainDataset.numInstances), evaluationMetric);
return learner;
} else {
return super.getLearningModule(name);
}