From b2324624121c91d49497699f13e400fd77d9cd89 Mon Sep 17 00:00:00 2001 From: Joshua Sloan Date: Mon, 8 Sep 2025 21:11:19 -0600 Subject: [PATCH] Added NumberOfLeaves option assignments Co-authored-by: isaiahleckie <40147803+isaiahleckie@users.noreply.github.com> --- .../SweepableEstimator/Estimators/FastForest.cs | 2 ++ 1 file changed, 2 insertions(+) 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,