diff --git a/src/Microsoft.ML.AutoML/SweepableEstimator/Estimators/FastForest.cs b/src/Microsoft.ML.AutoML/SweepableEstimator/Estimators/FastForest.cs index c9e900f023..56bb421c88 100644 --- a/src/Microsoft.ML.AutoML/SweepableEstimator/Estimators/FastForest.cs +++ b/src/Microsoft.ML.AutoML/SweepableEstimator/Estimators/FastForest.cs @@ -13,6 +13,7 @@ public override IEstimator BuildFromOption(MLContext context, Fast var option = new FastForestBinaryTrainer.Options() { NumberOfTrees = param.NumberOfTrees, + NumberOfLeaves = param.NumberOfLeaves, LabelColumnName = param.LabelColumnName, FeatureColumnName = param.FeatureColumnName, ExampleWeightColumnName = param.ExampleWeightColumnName, @@ -31,6 +32,7 @@ public override IEstimator BuildFromOption(MLContext context, Fast var option = new FastForestRegressionTrainer.Options() { NumberOfTrees = param.NumberOfTrees, + NumberOfLeaves = param.NumberOfLeaves, FeatureFraction = param.FeatureFraction, LabelColumnName = param.LabelColumnName, FeatureColumnName = param.FeatureColumnName,