diff --git a/src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs b/src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs index 2b7feb1dcf..aa5e8abc16 100644 --- a/src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs +++ b/src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs @@ -197,12 +197,6 @@ public static T TrainerKindApiValue(TrainerKinds trainerKind) public static bool IsTrainerOfKind(Type type, TrainerKinds trainerKind) { - if (type == typeof(Trainers.LogisticRegressionBinaryClassifier)) - return trainerKind == TrainerKinds.SignatureBinaryClassifierTrainer; - - if (type == typeof(Trainers.LogisticRegressionClassifier)) - return trainerKind == TrainerKinds.SignatureMultiClassClassifierTrainer; - if (trainerKind != TrainerKinds.SignatureMultiClassClassifierTrainer && trainerKind != TrainerKinds.SignatureMultiOutputRegressorTrainer) return type.Name.EndsWith(GetTrainerName(trainerKind));