107 references to RegressionPredictionTransformer
Microsoft.ML.AutoML (2)
TrainerExtensions\RegressionTrainerExtensions.cs (1)
91LightGbmRegressionTrainer.Options options = TrainerExtensionUtil.CreateLightGbmOptions<LightGbmRegressionTrainer.Options, float, RegressionPredictionTransformer<LightGbmRegressionModelParameters>, LightGbmRegressionModelParameters>(sweepParams, columnInfo);
Tuner\SmacTuner.cs (1)
148var model = trainer.Fit(trainTestSplit.TrainSet);
Microsoft.ML.Data (5)
Scorers\PredictionTransformer.cs (5)
574: base(Contracts.CheckRef(env, nameof(env)).Register(nameof(RegressionPredictionTransformer<TModel>)), model, inputSchema, featureColumn) 580: base(Contracts.CheckRef(env, nameof(env)).Register(nameof(RegressionPredictionTransformer<TModel>)), ctx) 609loaderAssemblyName: typeof(RegressionPredictionTransformer<>).Assembly.FullName); 756var host = Contracts.CheckRef(env, nameof(env)).Register(nameof(RegressionPredictionTransformer<IPredictorProducing<float>>)); 760Type predictionTransformerType = typeof(RegressionPredictionTransformer<>);
Microsoft.ML.EntryPoints (1)
PermutationFeatureImportance.cs (1)
209var pred = new RegressionPredictionTransformer<IPredictorProducing<float>>(
Microsoft.ML.FastTree (19)
FastTreeRegression.cs (4)
59: BoostingFastTreeTrainerBase<FastTreeRegressionTrainer.Options, RegressionPredictionTransformer<FastTreeRegressionModelParameters>, FastTreeRegressionModelParameters> 191private protected override RegressionPredictionTransformer<FastTreeRegressionModelParameters> MakeTransformer(FastTreeRegressionModelParameters model, DataViewSchema trainSchema) 196/// a <see cref="RegressionPredictionTransformer{FastTreeRegressionModelParameters}"/>. 198public RegressionPredictionTransformer<FastTreeRegressionModelParameters> Fit(IDataView trainData, IDataView validationData)
FastTreeTweedie.cs (4)
64: BoostingFastTreeTrainerBase<FastTreeTweedieTrainer.Options, RegressionPredictionTransformer<FastTreeTweedieModelParameters>, FastTreeTweedieModelParameters> 350private protected override RegressionPredictionTransformer<FastTreeTweedieModelParameters> MakeTransformer(FastTreeTweedieModelParameters model, DataViewSchema trainSchema) 355/// a <see cref="RegressionPredictionTransformer{FastTreeTweedieModelParameters}"/>. 357public RegressionPredictionTransformer<FastTreeTweedieModelParameters> Fit(IDataView trainData, IDataView validationData)
GamRegression.cs (4)
52public sealed class GamRegressionTrainer : GamTrainerBase<GamRegressionTrainer.Options, RegressionPredictionTransformer<GamRegressionModelParameters>, GamRegressionModelParameters> 125private protected override RegressionPredictionTransformer<GamRegressionModelParameters> MakeTransformer(GamRegressionModelParameters model, DataViewSchema trainSchema) 130/// a <see cref="RegressionPredictionTransformer{RegressionGamModelParameters}"/>. 132public RegressionPredictionTransformer<GamRegressionModelParameters> Fit(IDataView trainData, IDataView validationData)
RandomForestRegression.cs (4)
294: RandomForestTrainerBase<FastForestRegressionTrainer.Options, RegressionPredictionTransformer<FastForestRegressionModelParameters>, FastForestRegressionModelParameters> 494private protected override RegressionPredictionTransformer<FastForestRegressionModelParameters> MakeTransformer(FastForestRegressionModelParameters model, DataViewSchema trainSchema) 499/// a <see cref="RegressionPredictionTransformer{FastForestRegressionModelParameters}"/>. 501public RegressionPredictionTransformer<FastForestRegressionModelParameters> Fit(IDataView trainData, IDataView validationData)
TreeEnsembleFeaturizationEstimator.cs (3)
288var trained = trainer.Fit(input); 374var trained = trainer.Fit(input); 460var trained = trainer.Fit(input);
Microsoft.ML.IntegrationTests (15)
Explainability.cs (6)
45RegressionPredictionTransformer<FastTreeRegressionModelParameters> linearPredictor; 60linearPredictor = (loadedModel as TransformerChain<ITransformer>).LastTransformer as RegressionPredictionTransformer<FastTreeRegressionModelParameters>; 182var predictor = model.LastTransformer; 219var predictor = model.LastTransformer; 256var predictor = model.LastTransformer; 294var predictor = model.LastTransformer;
IntrospectiveTraining.cs (1)
300typeof(RegressionPredictionTransformer<FastForestRegressionModelParameters>)};
Training.cs (6)
329var firstModel = trainer.Fit(featurizedData); 333var firstModelPrime = trainer.Fit(featurizedData); 337var secondModel = trainer.Fit(featurizedData, firstModel.Model); 373var firstModel = trainer.Fit(featurizedData); 377var firstModelPrime = trainer.Fit(featurizedData); 381var secondModel = trainer.Fit(featurizedData, firstModel.Model);
Validation.cs (2)
49Assert.IsType<TransformerChain<RegressionPredictionTransformer<OlsModelParameters>>>(cvResult[0].Model); 121var trainedModel = mlContext.Regression.Trainers.FastTree(new FastTreeRegressionTrainer.Options
Microsoft.ML.LightGbm (4)
LightGbmRegressionTrainer.cs (4)
107RegressionPredictionTransformer<LightGbmRegressionModelParameters>, 254private protected override RegressionPredictionTransformer<LightGbmRegressionModelParameters> MakeTransformer(LightGbmRegressionModelParameters model, DataViewSchema trainSchema) 259/// a <see cref="RegressionPredictionTransformer{LightGbmRegressionModelParameters}"/>. 261public RegressionPredictionTransformer<LightGbmRegressionModelParameters> Fit(IDataView trainData, IDataView validationData)
Microsoft.ML.Mkl.Components (2)
OlsLinearRegression.cs (2)
67public sealed class OlsTrainer : TrainerEstimatorBase<RegressionPredictionTransformer<OlsModelParameters>, OlsModelParameters> 131private protected override RegressionPredictionTransformer<OlsModelParameters> MakeTransformer(OlsModelParameters model, DataViewSchema trainSchema)
Microsoft.ML.Samples (24)
Dynamic\Trainers\Regression\FastForest.cs (1)
35var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Regression\FastForestWithOptions.cs (1)
48var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Regression\FastTree.cs (1)
35var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Regression\FastTreeTweedie.cs (1)
35var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Regression\FastTreeTweedieWithOptions.cs (1)
50var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Regression\FastTreeWithOptions.cs (1)
51var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Regression\Gam.cs (1)
35var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Regression\GamWithOptions.cs (1)
46var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Regression\LbfgsPoissonRegression.cs (1)
33var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Regression\LbfgsPoissonRegressionWithOptions.cs (1)
47var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Regression\LightGbm.cs (1)
36var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Regression\LightGbmWithOptions.cs (1)
55var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Regression\OnlineGradientDescent.cs (1)
32var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Regression\OnlineGradientDescentWithOptions.cs (1)
47var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Regression\OrdinaryLeastSquares.cs (1)
32var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Regression\OrdinaryLeastSquaresAdvanced.cs (1)
55var model = pipeline.Fit(split.TrainSet);
Dynamic\Trainers\Regression\OrdinaryLeastSquaresWithOptions.cs (1)
44var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Regression\OrdinaryLeastSquaresWithOptionsAdvanced.cs (1)
59var model = pipeline.Fit(split.TrainSet);
Dynamic\Trainers\Regression\PermutationFeatureImportance.cs (1)
41var linearPredictor = model.LastTransformer;
Dynamic\Trainers\Regression\PermutationFeatureImportanceLoadFromDisk.cs (2)
48var linearPredictor = (model as TransformerChain<ITransformer>).LastTransformer as RegressionPredictionTransformer<OlsModelParameters>;
Dynamic\Trainers\Regression\Sdca.cs (1)
32var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Regression\SdcaWithOptions.cs (1)
48var model = pipeline.Fit(trainingData);
Dynamic\Transforms\CalculateFeatureContribution.cs (1)
39var linearModel = linearTrainer.Fit(transformedData);
Microsoft.ML.StandardTrainers (9)
Standard\Online\OnlineGradientDescent.cs (3)
58public sealed class OnlineGradientDescentTrainer : AveragedLinearTrainer<RegressionPredictionTransformer<LinearRegressionModelParameters>, LinearRegressionModelParameters> 129/// Trains a new <see cref="RegressionPredictionTransformer{LinearRegressionPredictor}"/>. 202private protected override RegressionPredictionTransformer<LinearRegressionModelParameters> MakeTransformer(LinearRegressionModelParameters model, DataViewSchema trainSchema)
Standard\PoissonRegression\PoissonRegression.cs (4)
58public sealed class LbfgsPoissonRegressionTrainer : LbfgsTrainerBase<LbfgsPoissonRegressionTrainer.Options, RegressionPredictionTransformer<PoissonRegressionModelParameters>, PoissonRegressionModelParameters> 127private protected override RegressionPredictionTransformer<PoissonRegressionModelParameters> MakeTransformer(PoissonRegressionModelParameters model, DataViewSchema trainSchema) 132/// a <see cref="RegressionPredictionTransformer{PoissonRegressionModelParameters}"/>. 134public RegressionPredictionTransformer<PoissonRegressionModelParameters> Fit(IDataView trainData, LinearModelParameters linearModel)
Standard\SdcaRegression.cs (2)
54public sealed class SdcaRegressionTrainer : SdcaTrainerBase<SdcaRegressionTrainer.Options, RegressionPredictionTransformer<LinearRegressionModelParameters>, LinearRegressionModelParameters> 201private protected override RegressionPredictionTransformer<LinearRegressionModelParameters> MakeTransformer(LinearRegressionModelParameters model, DataViewSchema trainSchema)
Microsoft.ML.Sweeper (1)
Algorithms\SmacSweeper.cs (1)
145var predictor = trainer.Fit(view);
Microsoft.ML.Tests (25)
FeatureContributionTests.cs (1)
31var model = ML.Regression.Trainers.Ols().Fit(data);
PermutationFeatureImportanceTests.cs (11)
38var model = ML.Regression.Trainers.OnlineGradientDescent().Fit(data); 49var castedModel = loadedModel as RegressionPredictionTransformer<LinearRegressionModelParameters>; 123var castedModel = lastTransformer as RegressionPredictionTransformer<LinearRegressionModelParameters>; 180var model = ML.Regression.Trainers.OnlineGradientDescent().Fit(data); 191var castedModel = loadedModel as RegressionPredictionTransformer<LinearRegressionModelParameters>; 268var model = ML.Regression.Trainers.OnlineGradientDescent().Fit(data); 279var castedModel = loadedModel as RegressionPredictionTransformer<LinearRegressionModelParameters>;
Scenarios\Api\CookbookSamples\CookbookSamplesDynamicApi.cs (1)
470var linearModel = model.LastTransformer;
TrainerEstimators\LbfgsTests.cs (1)
53var model = trainer.Fit(dataView);
TrainerEstimators\OlsLinearRegressionTests.cs (1)
21var model = trainer.Fit(dataView);
TrainerEstimators\OnlineLinearTests.cs (1)
31var ogdModel = ogdTrainer.Fit(regressionTrainData);
TrainerEstimators\TreeEstimators.cs (9)
220var model = trainer.Fit(dataView, dataView); 241var model = trainer.Fit(dataView, dataView); 260var model = trainer.Fit(dataView, dataView); 279var model = trainer.Fit(dataView, dataView); 298var model = trainer.Fit(dataView, dataView); 912var transformer = trainer.Fit(dataView); 930var transformer = trainer.Fit(dataView); 948var transformer = trainer.Fit(dataView); 973var transformer = trainer.Fit(dataView);