File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
src/Microsoft.ML/Runtime/EntryPoints Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -197,10 +197,10 @@ public static T TrainerKindApiValue<T>(TrainerKinds trainerKind)
197197
198198 public static bool IsTrainerOfKind ( Type type , TrainerKinds trainerKind )
199199 {
200- if ( type == typeof ( Trainers . BinaryLogisticRegressor ) )
200+ if ( type == typeof ( Trainers . LogisticRegressionBinaryClassifier ) )
201201 return trainerKind == TrainerKinds . SignatureBinaryClassifierTrainer ;
202202
203- if ( type == typeof ( Trainers . LogisticRegressor ) )
203+ if ( type == typeof ( Trainers . LogisticRegressionClassifier ) )
204204 return trainerKind == TrainerKinds . SignatureMultiClassClassifierTrainer ;
205205
206206 if ( trainerKind != TrainerKinds . SignatureMultiClassClassifierTrainer && trainerKind != TrainerKinds . SignatureMultiOutputRegressorTrainer )
You can’t perform that action at this time.
0 commit comments