We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 33c8364 commit 80a95b5Copy full SHA for 80a95b5
src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs
@@ -197,10 +197,10 @@ public static T TrainerKindApiValue<T>(TrainerKinds trainerKind)
197
198
public static bool IsTrainerOfKind(Type type, TrainerKinds trainerKind)
199
{
200
- if (type == typeof(Trainers.BinaryLogisticRegressor))
+ if (type == typeof(Trainers.LogisticRegressionBinaryClassifier))
201
return trainerKind == TrainerKinds.SignatureBinaryClassifierTrainer;
202
203
- if (type == typeof(Trainers.LogisticRegressor))
+ if (type == typeof(Trainers.LogisticRegressionClassifier))
204
return trainerKind == TrainerKinds.SignatureMultiClassClassifierTrainer;
205
206
if (trainerKind != TrainerKinds.SignatureMultiClassClassifierTrainer && trainerKind != TrainerKinds.SignatureMultiOutputRegressorTrainer)
0 commit comments