59 references to MulticlassPredictionTransformer
Microsoft.ML.AutoML (1)
TrainerExtensions\MultiTrainerExtensions.cs (1)
70LightGbmMulticlassTrainer.Options options = TrainerExtensionUtil.CreateLightGbmOptions<LightGbmMulticlassTrainer.Options, VBuffer<float>, MulticlassPredictionTransformer<OneVersusAllModelParameters>, OneVersusAllModelParameters>(sweepParams, columnInfo);
Microsoft.ML.Data (5)
Scorers\PredictionTransformer.cs (5)
486base(Contracts.CheckRef(env, nameof(env)).Register(nameof(MulticlassPredictionTransformer<TModel>)), model, inputSchema, featureColumn) 497: base(Contracts.CheckRef(env, nameof(env)).Register(nameof(MulticlassPredictionTransformer<TModel>)), ctx) 561loaderAssemblyName: typeof(MulticlassPredictionTransformer<>).Assembly.FullName); 739var host = Contracts.CheckRef(env, nameof(env)).Register(nameof(MulticlassPredictionTransformer<IPredictorProducing<VBuffer<float>>>)); 743Type predictionTransformerType = typeof(MulticlassPredictionTransformer<>);
Microsoft.ML.EntryPoints (1)
PermutationFeatureImportance.cs (1)
150var pred = new MulticlassPredictionTransformer<IPredictorProducing<VBuffer<float>>>(
Microsoft.ML.IntegrationTests (6)
IntrospectiveTraining.cs (3)
404var mcLrModel = (modelComponents[2] as TransformerChain<MulticlassPredictionTransformer<MaximumEntropyModelParameters>>).LastTransformer; 430private IEstimator<TransformerChain<MulticlassPredictionTransformer<MaximumEntropyModelParameters>>> StepTwo(MLContext mlContext)
Training.cs (3)
278var firstModel = trainer.Fit(featurizedData); 283var firstModelPrime = trainer.Fit(featurizedData); 288var secondModel = trainer.Fit(featurizedData, firstModel.Model);
Microsoft.ML.LightGbm (4)
LightGbmMulticlassTrainer.cs (4)
53MulticlassPredictionTransformer<OneVersusAllModelParameters>, 375private protected override MulticlassPredictionTransformer<OneVersusAllModelParameters> MakeTransformer(OneVersusAllModelParameters model, DataViewSchema trainSchema) 380/// a <see cref="MulticlassPredictionTransformer{OneVsAllModelParameters}"/>. 382public MulticlassPredictionTransformer<OneVersusAllModelParameters> Fit(IDataView trainData, IDataView validationData)
Microsoft.ML.PerformanceTests (4)
StochasticDualCoordinateAscentClassifierBench.cs (4)
36private TransformerChain<MulticlassPredictionTransformer<MaximumEntropyModelParameters>> _trainedModel; 57public TransformerChain<MulticlassPredictionTransformer<MaximumEntropyModelParameters>> TrainIris() => Train(_dataPath); 59private TransformerChain<MulticlassPredictionTransformer<MaximumEntropyModelParameters>> Train(string dataPath) 119var predicted = trainer.Fit(trans);
Microsoft.ML.Samples (3)
Dynamic\Trainers\MulticlassClassification\PermutationFeatureImportance.cs (1)
42var linearPredictor = model.LastTransformer;
Dynamic\Trainers\MulticlassClassification\PermutationFeatureImportanceLoadFromDisk.cs (2)
50var linearPredictor = (model as TransformerChain<ITransformer>).LastTransformer as MulticlassPredictionTransformer<MaximumEntropyModelParameters>;
Microsoft.ML.StandardTrainers (16)
Standard\LogisticRegression\MulticlassLogisticRegression.cs (4)
90MulticlassPredictionTransformer<MaximumEntropyModelParameters>, MaximumEntropyModelParameters> 383private protected override MulticlassPredictionTransformer<MaximumEntropyModelParameters> MakeTransformer(MaximumEntropyModelParameters model, DataViewSchema trainSchema) 388/// a <see cref="MulticlassPredictionTransformer{MulticlassLogisticRegressionModelParameters}"/>. 390public MulticlassPredictionTransformer<MaximumEntropyModelParameters> Fit(IDataView trainData, MaximumEntropyModelParameters modelParameters)
Standard\MulticlassClassification\MulticlassNaiveBayesTrainer.cs (2)
66public sealed class NaiveBayesMulticlassTrainer : TrainerEstimatorBase<MulticlassPredictionTransformer<NaiveBayesMulticlassModelParameters>, NaiveBayesMulticlassModelParameters> 129private protected override MulticlassPredictionTransformer<NaiveBayesMulticlassModelParameters> MakeTransformer(NaiveBayesMulticlassModelParameters model, DataViewSchema trainSchema)
Standard\MulticlassClassification\OneVersusAllTrainer.cs (4)
87public sealed class OneVersusAllTrainer : MetaMulticlassTrainer<MulticlassPredictionTransformer<OneVersusAllModelParameters>, OneVersusAllModelParameters> 207/// <summary> Trains a <see cref="MulticlassPredictionTransformer{OneVersusAllModelParameters}"/> model.</summary> 209/// <returns>A <see cref="MulticlassPredictionTransformer{OneVersusAllModelParameters}"/> model./></returns> 210public override MulticlassPredictionTransformer<OneVersusAllModelParameters> Fit(IDataView input)
Standard\MulticlassClassification\PairwiseCouplingTrainer.cs (3)
29using TTransformer = MulticlassPredictionTransformer<PairwiseCouplingModelParameters>; 76public sealed class PairwiseCouplingTrainer : MetaMulticlassTrainer<MulticlassPredictionTransformer<PairwiseCouplingModelParameters>, PairwiseCouplingModelParameters> 186public override TTransformer Fit(IDataView input)
Standard\SdcaMulticlass.cs (3)
80public abstract class SdcaMulticlassTrainerBase<TModel> : SdcaTrainerBase<SdcaMulticlassTrainerBase<TModel>.MulticlassOptions, MulticlassPredictionTransformer<TModel>, TModel> 564private protected override MulticlassPredictionTransformer<MaximumEntropyModelParameters> MakeTransformer( 661private protected override MulticlassPredictionTransformer<LinearMulticlassModelParameters> MakeTransformer(
Microsoft.ML.TensorFlow.Tests (1)
TensorflowTests.cs (1)
1887using var trainedModel = pipeline.Fit(trainDataset);
Microsoft.ML.Tests (15)
PermutationFeatureImportanceTests.cs (6)
525var model = ML.MulticlassClassification.Trainers.LbfgsMaximumEntropy().Fit(data); 536var castedModel = loadedModel as MulticlassPredictionTransformer<MaximumEntropyModelParameters>; 597var model = ML.MulticlassClassification.Trainers.LbfgsMaximumEntropy( 609var castedModel = loadedModel as MulticlassPredictionTransformer<MaximumEntropyModelParameters>;
Scenarios\OvaTest.cs (4)
39var model = pipeline.Fit(data); 75var model = pipeline.Fit(data); 110var model = pipeline.Fit(data); 144var model = pipeline.Fit(data);
TrainerEstimators\LbfgsTests.cs (3)
41var model = trainer.Fit(transformedDataView); 223var lastTransformer = ((TransformerChain<ITransformer>)transformerChain).LastTransformer as MulticlassPredictionTransformer<MaximumEntropyModelParameters>;
TrainerEstimators\TreeEstimators.cs (2)
374var model = trainer.Fit(transformedDataView, transformedDataView); 474var gbm = gbmTrainer.Fit(dataView);
Microsoft.ML.Vision (3)
ImageClassificationTrainer.cs (3)
79TrainerEstimatorBase<MulticlassPredictionTransformer<ImageClassificationModelParameters>, 649private protected override MulticlassPredictionTransformer<ImageClassificationModelParameters> MakeTransformer( 1341public MulticlassPredictionTransformer<ImageClassificationModelParameters> Fit(