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