Skip to content

Commit 3102180

Browse files
authored
Remove special case for Logistic Regression in MacroUtils.cs (#147)
1 parent 80a95b5 commit 3102180

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs

-6
Original file line numberDiff line numberDiff line change
@@ -197,12 +197,6 @@ public static T TrainerKindApiValue<T>(TrainerKinds trainerKind)
197197

198198
public static bool IsTrainerOfKind(Type type, TrainerKinds trainerKind)
199199
{
200-
if (type == typeof(Trainers.LogisticRegressionBinaryClassifier))
201-
return trainerKind == TrainerKinds.SignatureBinaryClassifierTrainer;
202-
203-
if (type == typeof(Trainers.LogisticRegressionClassifier))
204-
return trainerKind == TrainerKinds.SignatureMultiClassClassifierTrainer;
205-
206200
if (trainerKind != TrainerKinds.SignatureMultiClassClassifierTrainer && trainerKind != TrainerKinds.SignatureMultiOutputRegressorTrainer)
207201
return type.Name.EndsWith(GetTrainerName(trainerKind));
208202

0 commit comments

Comments
 (0)