31 references to RankingPredictionTransformer
Microsoft.ML.AutoML (1)
TrainerExtensions\RankingTrainerExtensions.cs (1)
25float, RankingPredictionTransformer<LightGbmRankingModelParameters>, LightGbmRankingModelParameters>(sweepParams, columnInfo);
Microsoft.ML.Data (5)
Scorers\PredictionTransformer.cs (5)
622: base(Contracts.CheckRef(env, nameof(env)).Register(nameof(RankingPredictionTransformer<TModel>)), model, inputSchema, featureColumn) 628: base(Contracts.CheckRef(env, nameof(env)).Register(nameof(RankingPredictionTransformer<TModel>)), ctx) 657loaderAssemblyName: typeof(RankingPredictionTransformer<>).Assembly.FullName); 774var host = Contracts.CheckRef(env, nameof(env)).Register(nameof(RankingPredictionTransformer<IPredictorProducing<float>>)); 778Type predictionTransformerType = typeof(RankingPredictionTransformer<>);
Microsoft.ML.EntryPoints (1)
PermutationFeatureImportance.cs (1)
262var pred = new RankingPredictionTransformer<IPredictorProducing<float>>(
Microsoft.ML.FastTree (5)
FastTreeRanking.cs (4)
68: BoostingFastTreeTrainerBase<FastTreeRankingTrainer.Options, RankingPredictionTransformer<FastTreeRankingModelParameters>, FastTreeRankingModelParameters> 475private protected override RankingPredictionTransformer<FastTreeRankingModelParameters> MakeTransformer(FastTreeRankingModelParameters model, DataViewSchema trainSchema) 480/// a <see cref="RankingPredictionTransformer{FastTreeRankingModelParameters}"/>. 482public RankingPredictionTransformer<FastTreeRankingModelParameters> Fit(IDataView trainData, IDataView validationData)
TreeEnsembleFeaturizationEstimator.cs (1)
417var trained = trainer.Fit(input);
Microsoft.ML.LightGbm (4)
LightGbmRankingTrainer.cs (4)
105RankingPredictionTransformer<LightGbmRankingModelParameters>, 306private protected override RankingPredictionTransformer<LightGbmRankingModelParameters> MakeTransformer(LightGbmRankingModelParameters model, DataViewSchema trainSchema) 311/// a <see cref="RankingPredictionTransformer{LightGbmRankingModelParameters}"/>. 313public RankingPredictionTransformer<LightGbmRankingModelParameters> Fit(IDataView trainData, IDataView validationData)
Microsoft.ML.Samples (7)
Dynamic\Trainers\Ranking\FastTree.cs (1)
33var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Ranking\FastTreeWithOptions.cs (1)
47var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Ranking\LightGbm.cs (1)
33var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Ranking\LightGbmWithOptions.cs (1)
48var model = pipeline.Fit(trainingData);
Dynamic\Trainers\Ranking\PermutationFeatureImportance.cs (1)
42var linearPredictor = model.LastTransformer;
Dynamic\Trainers\Ranking\PermutationFeatureImportanceLoadFromDisk.cs (2)
49var linearPredictor = (model as TransformerChain<ITransformer>).LastTransformer as RankingPredictionTransformer<FastTreeRankingModelParameters>;
Microsoft.ML.Tests (8)
PermutationFeatureImportanceTests.cs (6)
676var model = ML.Ranking.Trainers.FastTree().Fit(data); 687var castedModel = loadedModel as RankingPredictionTransformer<FastTreeRankingModelParameters>; 745var model = ML.Ranking.Trainers.FastTree().Fit(data); 756var castedModel = loadedModel as RankingPredictionTransformer<FastTreeRankingModelParameters>;
TrainerEstimators\TreeEstimators.cs (2)
187var model = trainer.Fit(transformedDataView, transformedDataView); 205var model = trainer.Fit(transformedDataView, transformedDataView);