Skip to content

Commit 6a4df7c

Browse files
ganikTomFinley
authored andcommitted
Allow only BinaryPredictionTransformer estimators in OVA (#2949)
1 parent 2ee4cab commit 6a4df7c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/Microsoft.ML.StandardTrainers/StandardTrainersCatalog.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -632,7 +632,7 @@ private static ICalibratorTrainer GetCalibratorTrainerOrThrow(IExceptionContext
632632
/// <param name="useProbabilities">Use probabilities (vs. raw outputs) to identify top-score category.</param>
633633
/// <typeparam name="TModel">The type of the model. This type parameter will usually be inferred automatically from <paramref name="binaryEstimator"/>.</typeparam>
634634
public static OneVersusAllTrainer OneVersusAll<TModel>(this MulticlassClassificationCatalog.MulticlassClassificationTrainers catalog,
635-
ITrainerEstimator<ISingleFeaturePredictionTransformer<TModel>, TModel> binaryEstimator,
635+
ITrainerEstimator<BinaryPredictionTransformer<TModel>, TModel> binaryEstimator,
636636
string labelColumnName = DefaultColumnNames.Label,
637637
bool imputeMissingLabelsAsNegative = false,
638638
IEstimator<ISingleFeaturePredictionTransformer<ICalibrator>> calibrator = null,

0 commit comments

Comments
 (0)