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