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