49 references to BinaryClassification
Microsoft.ML.Data (4)
DataLoadSave\TransformerChain.cs (1)
300if (predictor.PredictionKind == PredictionKind.BinaryClassification)
Scorers\SchemaBindablePredictorWrapper.cs (3)
364case PredictionKind.BinaryClassification: 494if (Predictor.PredictionKind != PredictionKind.BinaryClassification) 495ch.Warning("Scoring predictor of kind '{0}' as '{1}'.", Predictor.PredictionKind, PredictionKind.BinaryClassification);
Microsoft.ML.Ensemble (8)
PipelineEnsemble.cs (1)
296public override PredictionKind PredictionKind { get { return PredictionKind.BinaryClassification; } }
Selector\SubModelSelector\AllSelector.cs (1)
21protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;
Selector\SubModelSelector\BaseSubModelSelector.cs (2)
61case PredictionKind.BinaryClassification: 107case PredictionKind.BinaryClassification:
Selector\SubModelSelector\BestDiverseSelectorBinary.cs (1)
48protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;
Selector\SubModelSelector\BestPerformanceSelector.cs (1)
51protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;
Trainer\Binary\EnsembleTrainer.cs (2)
78Host.CheckParam(predictionKind == PredictionKind.BinaryClassification, nameof(PredictionKind)); 81private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;
Microsoft.ML.EntryPoints (2)
PermutationFeatureImportance.cs (1)
68if (predictor.PredictionKind == PredictionKind.BinaryClassification)
ScoreColumnSelector.cs (1)
73if (input.PredictorModel.Predictor.PredictionKind == PredictionKind.BinaryClassification)
Microsoft.ML.FastTree (10)
FastTreeClassification.cs (2)
100private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification; 182private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;
GamClassification.cs (2)
79private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification; 200private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;
GamModelParameters.cs (1)
902case PredictionKind.BinaryClassification:
RandomForestClassification.cs (2)
89private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification; 175private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;
TreeEnsemble\TreeEnsembleCombiner.cs (3)
26case PredictionKind.BinaryClassification: 32throw _host.ExceptUserArg(nameof(kind), $"Tree ensembles can be either of type {nameof(PredictionKind.BinaryClassification)}, " + 102case PredictionKind.BinaryClassification:
Microsoft.ML.LightGbm (2)
LightGbmBinaryTrainer.cs (2)
57private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification; 123private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;
Microsoft.ML.Mkl.Components (1)
SymSgdClassificationTrainer.cs (1)
235private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;
Microsoft.ML.OnnxConverter (1)
SaveOnnxCommand.cs (1)
359if (rawPred.PredictionKind == PredictionKind.BinaryClassification || rawPred.PredictionKind == PredictionKind.MulticlassClassification)
Microsoft.ML.Predictor.Tests (8)
TestPredictors.cs (8)
669IModelCombiner combiner = new TreeEnsembleCombiner(Env, PredictionKind.BinaryClassification); 795CombineAndTestEnsembles(dataView, "pe", "oc=average", PredictionKind.BinaryClassification, predictors); 852if (predictionKind == PredictionKind.BinaryClassification) 868if (predictionKind == PredictionKind.BinaryClassification) 896var probGetter = predictionKind == PredictionKind.BinaryClassification ? 899var predGetter = predictionKind == PredictionKind.BinaryClassification ? 915probGetters[i] = predictionKind == PredictionKind.BinaryClassification ? 918predGetters[i] = predictionKind == PredictionKind.BinaryClassification ?
Microsoft.ML.StandardTrainers (13)
FactorizationMachine\FactorizationMachineTrainer.cs (1)
180PredictionKind ITrainer.PredictionKind => PredictionKind.BinaryClassification;
FactorizationMachine\FieldAwareFactorizationMachineModelParameters.cs (1)
30private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;
LdSvm\LdSvmModelParameters.cs (1)
156private protected override PredictionKind PredictionKind { get { return PredictionKind.BinaryClassification; } }
LdSvm\LdSvmTrainer.cs (1)
178private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;
Standard\LinearModelParameters.cs (1)
510private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;
Standard\LogisticRegression\LogisticRegression.cs (1)
161private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;
Standard\Online\AveragedPerceptron.cs (1)
192private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;
Standard\Online\LinearSvm.cs (1)
303private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;
Standard\SdcaBinary.cs (2)
1457private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification; 1951private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;
Standard\Simple\SimpleTrainers.cs (3)
61private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification; 209PredictionKind ITrainer.PredictionKind => PredictionKind.BinaryClassification; 438private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;