From 9ce3e8cf40dbbfed7502cedcbcf1874c50ba2af4 Mon Sep 17 00:00:00 2001 From: Eric Erhardt Date: Mon, 14 May 2018 11:00:25 -0500 Subject: [PATCH] Fix build break The previous 2 changes conflicted. Resovling the break that happened between them. --- src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs b/src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs index 8c5613014e..2b7feb1dcf 100644 --- a/src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs +++ b/src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs @@ -197,10 +197,10 @@ public static T TrainerKindApiValue(TrainerKinds trainerKind) public static bool IsTrainerOfKind(Type type, TrainerKinds trainerKind) { - if (type == typeof(Trainers.BinaryLogisticRegressor)) + if (type == typeof(Trainers.LogisticRegressionBinaryClassifier)) return trainerKind == TrainerKinds.SignatureBinaryClassifierTrainer; - if (type == typeof(Trainers.LogisticRegressor)) + if (type == typeof(Trainers.LogisticRegressionClassifier)) return trainerKind == TrainerKinds.SignatureMultiClassClassifierTrainer; if (trainerKind != TrainerKinds.SignatureMultiClassClassifierTrainer && trainerKind != TrainerKinds.SignatureMultiOutputRegressorTrainer)