3 instantiations of FastTreeRankingModelParameters
Microsoft.ML.FastTree (3)
FastTreeRanking.cs (2)
152
return new
FastTreeRankingModelParameters
(Host, TrainedEnsemble, FeatureCount, InnerOptions);
1071
return new
FastTreeRankingModelParameters
(env, ctx);
TreeEnsemble\TreeEnsembleCombiner.cs (1)
112
return new
FastTreeRankingModelParameters
(_host, ensemble, featureCount, null);
17 references to FastTreeRankingModelParameters
Microsoft.ML.FastTree (11)
FastTreeRanking.cs (11)
33
[assembly: LoadableClass(typeof(
FastTreeRankingModelParameters
), null, typeof(SignatureLoadModel),
35
FastTreeRankingModelParameters
.LoaderSignature)]
68
: BoostingFastTreeTrainerBase<FastTreeRankingTrainer.Options, RankingPredictionTransformer<
FastTreeRankingModelParameters
>,
FastTreeRankingModelParameters
>
138
private protected override
FastTreeRankingModelParameters
TrainModelCore(TrainContext context)
475
private protected override RankingPredictionTransformer<
FastTreeRankingModelParameters
> MakeTransformer(
FastTreeRankingModelParameters
model, DataViewSchema trainSchema)
476
=> new RankingPredictionTransformer<
FastTreeRankingModelParameters
>(Host, model, trainSchema, FeatureColumn.Name);
482
public RankingPredictionTransformer<
FastTreeRankingModelParameters
> Fit(IDataView trainData, IDataView validationData)
1044
loaderAssemblyName: typeof(
FastTreeRankingModelParameters
).Assembly.FullName);
1069
internal static
FastTreeRankingModelParameters
Create(IHostEnvironment env, ModelLoadContext ctx)
Microsoft.ML.PerformanceTests (3)
Numeric\Ranking.cs (3)
45
var environment = EnvironmentFactory.CreateRankingEnvironment<RankingEvaluator, TextLoader, HashingTransformer, FastTreeRankingTrainer,
FastTreeRankingModelParameters
>();
99
var environment = EnvironmentFactory.CreateRankingEnvironment<RankingEvaluator, TextLoader, HashingTransformer, FastTreeRankingTrainer,
FastTreeRankingModelParameters
>();
109
var environment = EnvironmentFactory.CreateRankingEnvironment<RankingEvaluator, TextLoader, HashingTransformer, FastTreeRankingTrainer,
FastTreeRankingModelParameters
>();
Microsoft.ML.Samples (1)
Dynamic\Trainers\Ranking\PermutationFeatureImportanceLoadFromDisk.cs (1)
49
var linearPredictor = (model as TransformerChain<ITransformer>).LastTransformer as RankingPredictionTransformer<
FastTreeRankingModelParameters
>;
Microsoft.ML.Tests (2)
PermutationFeatureImportanceTests.cs (2)
687
var castedModel = loadedModel as RankingPredictionTransformer<
FastTreeRankingModelParameters
>;
756
var castedModel = loadedModel as RankingPredictionTransformer<
FastTreeRankingModelParameters
>;