From f8160906ac204c4aebae16a06d41ced2c0ddc37e Mon Sep 17 00:00:00 2001 From: Yael Dekel Date: Mon, 14 May 2018 09:28:26 -0700 Subject: [PATCH] Remove special case for Logistic Regression in MacroUtils.cs --- src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs | 6 ------ 1 file changed, 6 deletions(-) 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));