change LabeledModelTrainer