Remove special case for Logistic Regression in MacroUtils.cs (#147)

dotnet · May 14, 2018 · 3102180 · 3102180
1 parent 80a95b5
commit 3102180
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs b/src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs
@@ -197,12 +197,6 @@ public static T TrainerKindApiValue<T>(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));