2 instantiations of TreeEnsembleFeaturizationTransformer
Microsoft.ML.FastTree (2)
TreeEnsembleFeaturizationEstimator.cs (1)
115
return new
TreeEnsembleFeaturizationTransformer
(Env, input.Schema, input.Schema[FeatureColumnName], model,
TreeEnsembleFeaturizationTransformer.cs (1)
190
=> new
TreeEnsembleFeaturizationTransformer
(env, ctx);
21 references to TreeEnsembleFeaturizationTransformer
Microsoft.ML.FastTree (11)
TreeEnsembleFeaturizationEstimator.cs (3)
20
public abstract class TreeEnsembleFeaturizationEstimatorBase : IEstimator<
TreeEnsembleFeaturizationTransformer
>
105
/// <returns>The trees used in <see cref="
TreeEnsembleFeaturizationTransformer
"/>.</returns>
112
public
TreeEnsembleFeaturizationTransformer
Fit(IDataView input)
TreeEnsembleFeaturizationTransformer.cs (8)
12
[assembly: LoadableClass(typeof(
TreeEnsembleFeaturizationTransformer
), typeof(
TreeEnsembleFeaturizationTransformer
),
13
null, typeof(SignatureLoadModel), "",
TreeEnsembleFeaturizationTransformer
.LoaderSignature)]
72
base(Contracts.CheckRef(env, nameof(env)).Register(nameof(
TreeEnsembleFeaturizationTransformer
)), modelParameters, inputSchema)
105
: base(Contracts.CheckRef(host, nameof(host)).Register(nameof(
TreeEnsembleFeaturizationTransformer
)), ctx)
139
/// <see cref="
TreeEnsembleFeaturizationTransformer
"/> appends three columns to the <paramref name="inputSchema"/>.
186
loaderAssemblyName: typeof(
TreeEnsembleFeaturizationTransformer
).Assembly.FullName);
189
internal static
TreeEnsembleFeaturizationTransformer
Create(IHostEnvironment env, ModelLoadContext ctx)
Microsoft.ML.Samples (7)
Dynamic\Transforms\TreeFeaturization\FastForestBinaryFeaturizationWithOptions.cs (1)
75
var
model = pipeline.Fit(dataView);
Dynamic\Transforms\TreeFeaturization\FastForestRegressionFeaturizationWithOptions.cs (1)
75
var
model = pipeline.Fit(dataView);
Dynamic\Transforms\TreeFeaturization\FastTreeBinaryFeaturizationWithOptions.cs (1)
77
var
model = pipeline.Fit(dataView);
Dynamic\Transforms\TreeFeaturization\FastTreeRankingFeaturizationWithOptions.cs (1)
73
var
model = pipeline.Fit(dataView);
Dynamic\Transforms\TreeFeaturization\FastTreeRegressionFeaturizationWithOptions.cs (1)
75
var
model = pipeline.Fit(dataView);
Dynamic\Transforms\TreeFeaturization\FastTreeTweedieFeaturizationWithOptions.cs (1)
75
var
model = pipeline.Fit(dataView);
Dynamic\Transforms\TreeFeaturization\PretrainedTreeEnsembleFeaturizationWithOptions.cs (1)
65
var
treeFeaturizer = mlContext.Transforms
Microsoft.ML.Tests (3)
TrainerEstimators\TreeEnsembleFeaturizerTest.cs (3)
149
var
treeFeaturizer = new TreeEnsembleFeaturizationTransformer(ML, dataView.Schema, dataView.Schema["Features"], model.Model.SubModel,
200
var
treeFeaturizer = new TreeEnsembleFeaturizationTransformer(ML, dataView.Schema, dataView.Schema["Features"], model.Model,
265
var
treeFeaturizer = ML.Transforms.FeaturizeByPretrainTreeEnsemble(options).Fit(dataView);