31 references to MulticlassClassification
Microsoft.ML.Data (2)
DataLoadSave\TransformerChain.cs (1)
303else if (predictor.PredictionKind == PredictionKind.MulticlassClassification)
Scorers\SchemaBindablePredictorWrapper.cs (1)
366case PredictionKind.MulticlassClassification:
Microsoft.ML.Ensemble (9)
PipelineEnsemble.cs (1)
269return PredictionKind.MulticlassClassification;
Selector\SubModelSelector\AllSelectorMulticlass.cs (1)
23protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification;
Selector\SubModelSelector\BaseSubModelSelector.cs (2)
65case PredictionKind.MulticlassClassification: 124case PredictionKind.MulticlassClassification:
Selector\SubModelSelector\BestDiverseSelectorMulticlass.cs (1)
40protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification;
Selector\SubModelSelector\BestPerformanceSelectorMulticlass.cs (1)
47protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification;
Trainer\Multiclass\EnsembleMulticlassModelParameters.cs (1)
108private protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification;
Trainer\Multiclass\MulticlassDataPartitionEnsembleTrainer.cs (2)
85Host.CheckParam(predictionKind == PredictionKind.MulticlassClassification, nameof(PredictionKind)); 88private protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification;
Microsoft.ML.EntryPoints (1)
PermutationFeatureImportance.cs (1)
70else if (predictor.PredictionKind == PredictionKind.MulticlassClassification)
Microsoft.ML.LightGbm (3)
LightGbmMulticlassTrainer.cs (2)
69private protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification; 337ch.Assert(PredictionKind == PredictionKind.MulticlassClassification);
LightGbmTrainerBase.cs (1)
728ch.Assert(((ITrainer)this).PredictionKind != PredictionKind.MulticlassClassification || GbmOptions.ContainsKey("num_class"),
Microsoft.ML.OnnxConverter (1)
SaveOnnxCommand.cs (1)
359if (rawPred.PredictionKind == PredictionKind.BinaryClassification || rawPred.PredictionKind == PredictionKind.MulticlassClassification)
Microsoft.ML.Predictor.Tests (5)
TestPredictors.cs (5)
832CombineAndTestEnsembles(dataView, "weightedensemblemulticlass", "oc=multiaverage", PredictionKind.MulticlassClassification, predictors); 890var scoreGetter = predictionKind == PredictionKind.MulticlassClassification ? 893var vectorScoreGetter = predictionKind == PredictionKind.MulticlassClassification ? 909scoreGetters[i] = predictionKind == PredictionKind.MulticlassClassification ? 912vectorScoreGetters[i] = predictionKind == PredictionKind.MulticlassClassification ?
Microsoft.ML.StandardTrainers (8)
Standard\LogisticRegression\MulticlassLogisticRegression.cs (2)
167private protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification; 447private protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification;
Standard\MulticlassClassification\MetaMulticlassTrainer.cs (1)
48private protected PredictionKind PredictionKind => PredictionKind.MulticlassClassification;
Standard\MulticlassClassification\MulticlassNaiveBayesTrainer.cs (2)
78private protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification; 252private protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification;
Standard\MulticlassClassification\OneVersusAllTrainer.cs (1)
278private protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification;
Standard\MulticlassClassification\PairwiseCouplingTrainer.cs (1)
256private protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification;
Standard\SdcaMulticlass.cs (1)
113private protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification;
Microsoft.ML.Vision (2)
ImageClassificationTrainer.cs (2)
460private protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification; 1380private protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification;