2 types derived from TreeEnsembleModelParameters
Microsoft.ML.FastTree (2)
FastTree.cs (2)
3362public abstract class TreeEnsembleModelParametersBasedOnRegressionTree : TreeEnsembleModelParameters, ICanGetSummaryAsIDataView 3408public abstract class TreeEnsembleModelParametersBasedOnQuantileRegressionTree : TreeEnsembleModelParameters, ICanGetSummaryAsIDataView
47 references to TreeEnsembleModelParameters
Microsoft.ML.AutoML (4)
Sweepers\SmacSweeper.cs (4)
347private static readonly MethodInfo _getLeafMethod = typeof(TreeEnsembleModelParameters).GetMethod("GetLeaf", BindingFlags.NonPublic | BindingFlags.Instance); 348private static readonly MethodInfo _getLeafValueMethod = typeof(TreeEnsembleModelParameters).GetMethod("GetLeafValue", BindingFlags.NonPublic | BindingFlags.Instance); 350private static int GetLeaf(TreeEnsembleModelParameters model, int treeId, VBuffer<Float> features) 356private static float GetLeafValue(TreeEnsembleModelParameters model, int treeId, int leafId)
Microsoft.ML.FastTree (43)
FastTree.cs (4)
3354/// <see cref="TreeEnsembleModelParameters"/> plus a strongly-typed public attribute, 3357/// called to create <see cref="TrainedTreeEnsemble"/> inside <see cref="TreeEnsembleModelParameters"/>. 3400/// <see cref="TreeEnsembleModelParameters"/> plus a strongly-typed public attribute, 3403/// called to create <see cref="TrainedTreeEnsemble"/> inside <see cref="TreeEnsembleModelParameters"/>.
TreeEnsemble\TreeEnsembleCombiner.cs (2)
61var tree = predictor as TreeEnsembleModelParameters;
TreeEnsembleFeaturizationEstimator.cs (22)
101/// All derived class should implement <see cref="PrepareModel(IDataView)"/> to tell how to get a <see cref="TreeEnsembleModelParameters"/> 106private protected abstract TreeEnsembleModelParameters PrepareModel(IDataView input); 109/// Produce a <see cref="TreeEnsembleModelParameters"/> which maps the column called <see cref="OptionsBase.InputColumnName"/> in <paramref name="input"/> 114var model = PrepareModel(input); 152/// A <see cref="IEstimator{TTransformer}"/> which contains a pre-trained <see cref="TreeEnsembleModelParameters"/> and calling its 188/// The pretrained tree model used to do tree-based featurization. Note that <see cref="TreeEnsembleModelParameters"/> contains a collection of decision trees. 190public TreeEnsembleModelParameters ModelParameters; 193private readonly TreeEnsembleModelParameters _modelParameters; 201/// Produce the <see cref="TreeEnsembleModelParameters"/> for tree-based feature engineering. This function does not 204private protected override TreeEnsembleModelParameters PrepareModel(IDataView input) => _modelParameters; 231/// The configuration of <see cref="FastTreeBinaryTrainer"/> used to train the underlying <see cref="TreeEnsembleModelParameters"/>. 242private protected override TreeEnsembleModelParameters PrepareModel(IDataView input) 274/// The configuration of <see cref="FastTreeRegressionTrainer"/> used to train the underlying <see cref="TreeEnsembleModelParameters"/>. 285private protected override TreeEnsembleModelParameters PrepareModel(IDataView input) 317/// The configuration of <see cref="FastForestBinaryTrainer"/> used to train the underlying <see cref="TreeEnsembleModelParameters"/>. 328private protected override TreeEnsembleModelParameters PrepareModel(IDataView input) 360/// The configuration of <see cref="FastForestRegressionTrainer"/> used to train the underlying <see cref="TreeEnsembleModelParameters"/>. 371private protected override TreeEnsembleModelParameters PrepareModel(IDataView input) 403/// The configuration of <see cref="FastTreeRankingTrainer"/> used to train the underlying <see cref="TreeEnsembleModelParameters"/>. 414private protected override TreeEnsembleModelParameters PrepareModel(IDataView input) 446/// The configuration of <see cref="FastTreeTweedieTrainer"/> used to train the underlying <see cref="TreeEnsembleModelParameters"/>. 457private protected override TreeEnsembleModelParameters PrepareModel(IDataView input)
TreeEnsembleFeaturizationTransformer.cs (2)
22public sealed class TreeEnsembleFeaturizationTransformer : PredictionTransformerBase<TreeEnsembleModelParameters> 70DataViewSchema.Column featureColumn, TreeEnsembleModelParameters modelParameters,
TreeEnsembleFeaturizer.cs (6)
228private readonly TreeEnsembleModelParameters _ensemble; 243public State(IExceptionContext ectx, DataViewRow input, TreeEnsembleModelParameters ensemble, int numLeaves, int featureIndex) 391private readonly TreeEnsembleModelParameters _ensemble; 417_ensemble = predictor as TreeEnsembleModelParameters; 436ctx.LoadModel<TreeEnsembleModelParameters, SignatureLoadModel>(env, out _ensemble, "Ensemble"); 462private static int CountLeaves(TreeEnsembleModelParameters ensemble)
TreeTrainersCatalog.cs (7)
441/// Create <see cref="PretrainedTreeFeaturizationEstimator"/>, which produces tree-based features given a <see cref="TreeEnsembleModelParameters"/>. 462/// Create <see cref="FastForestRegressionFeaturizationEstimator"/>, which uses <see cref="FastForestRegressionTrainer"/> to train <see cref="TreeEnsembleModelParameters"/> to create tree-based features. 483/// Create <see cref="FastTreeRegressionFeaturizationEstimator"/>, which uses <see cref="FastTreeRegressionTrainer"/> to train <see cref="TreeEnsembleModelParameters"/> to create tree-based features. 504/// Create <see cref="FastForestBinaryFeaturizationEstimator"/>, which uses <see cref="FastForestBinaryTrainer"/> to train <see cref="TreeEnsembleModelParameters"/> to create tree-based features. 525/// Create <see cref="FastTreeBinaryFeaturizationEstimator"/>, which uses <see cref="FastTreeBinaryTrainer"/> to train <see cref="TreeEnsembleModelParameters"/> to create tree-based features. 546/// Create <see cref="FastTreeRankingFeaturizationEstimator"/>, which uses <see cref="FastTreeRankingTrainer"/> to train <see cref="TreeEnsembleModelParameters"/> to create tree-based features. 567/// Create <see cref="FastTreeTweedieFeaturizationEstimator"/>, which uses <see cref="FastTreeTweedieTrainer"/> to train <see cref="TreeEnsembleModelParameters"/> to create tree-based features.