Skip to content

Commit 6a5ec43

Browse files
Added NumberOfLeaves option assignments (#7499)
Co-authored-by: isaiahleckie <[email protected]>
1 parent d76daf6 commit 6a5ec43

File tree

1 file changed

+2
-0
lines changed
  • src/Microsoft.ML.AutoML/SweepableEstimator/Estimators

1 file changed

+2
-0
lines changed

src/Microsoft.ML.AutoML/SweepableEstimator/Estimators/FastForest.cs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ public override IEstimator<ITransformer> BuildFromOption(MLContext context, Fast
1313
var option = new FastForestBinaryTrainer.Options()
1414
{
1515
NumberOfTrees = param.NumberOfTrees,
16+
NumberOfLeaves = param.NumberOfLeaves,
1617
LabelColumnName = param.LabelColumnName,
1718
FeatureColumnName = param.FeatureColumnName,
1819
ExampleWeightColumnName = param.ExampleWeightColumnName,
@@ -31,6 +32,7 @@ public override IEstimator<ITransformer> BuildFromOption(MLContext context, Fast
3132
var option = new FastForestRegressionTrainer.Options()
3233
{
3334
NumberOfTrees = param.NumberOfTrees,
35+
NumberOfLeaves = param.NumberOfLeaves,
3436
FeatureFraction = param.FeatureFraction,
3537
LabelColumnName = param.LabelColumnName,
3638
FeatureColumnName = param.FeatureColumnName,

0 commit comments

Comments
 (0)