in preparation to remove one of the constructors for FastTreeTrainerBase

Author: abgoswam

src/Microsoft.ML.FastTree/TreeTrainersCatalog.cs

@@ -4,6 +4,7 @@
 
 using System;
 using Microsoft.ML.Data;
+using Microsoft.ML.EntryPoints;
 using Microsoft.ML.Trainers.FastTree;
 
 namespace Microsoft.ML
@@ -40,13 +41,15 @@ public static FastTreeRegressionTrainer FastTree(this RegressionContext.Regressi
             {
                 LabelColumn = labelColumn,
                 FeatureColumn = featureColumn,
-                WeightColumn = weights,
                 NumLeaves = numLeaves,
                 NumTrees = numTrees,
                 MinDocumentsInLeafs = minDatapointsInLeaves,
                 LearningRates = learningRate,
             };
 
+            if (weights != null)
+                options.WeightColumn = Optional<string>.Explicit(weights);
+
             return new FastTreeRegressionTrainer(env, options);
         }
 
@@ -90,13 +93,15 @@ public static FastTreeBinaryClassificationTrainer FastTree(this BinaryClassifica
             {
                 LabelColumn = labelColumn,
                 FeatureColumn = featureColumn,
-                WeightColumn = weights,
                 NumLeaves = numLeaves,
                 NumTrees = numTrees,
                 MinDocumentsInLeafs = minDatapointsInLeaves,
                 LearningRates = learningRate,
             };
 
+            if (weights != null)
+                options.WeightColumn = Optional<string>.Explicit(weights);
+
             return new FastTreeBinaryClassificationTrainer(env, options);
         }
 
@@ -230,13 +235,15 @@ public static FastTreeTweedieTrainer FastTreeTweedie(this RegressionContext.Regr
             {
                 LabelColumn = labelColumn,
                 FeatureColumn = featureColumn,
-                WeightColumn = weights,
                 NumLeaves = numLeaves,
                 NumTrees = numTrees,
                 MinDocumentsInLeafs = minDatapointsInLeaves,
                 LearningRates = learningRate,
             };
 
+            if (weights != null)
+                options.WeightColumn = Optional<string>.Explicit(weights);
+
             return new FastTreeTweedieTrainer(env, options);
         }
 
@@ -278,12 +285,14 @@ public static FastForestRegression FastForest(this RegressionContext.RegressionT
             {
                 LabelColumn = labelColumn,
                 FeatureColumn = featureColumn,
-                WeightColumn = weights,
                 NumLeaves = numLeaves,
                 NumTrees = numTrees,
                 MinDocumentsInLeafs = minDatapointsInLeaves,
             };
 
+            if (weights != null)
+                options.WeightColumn = Optional<string>.Explicit(weights);
+
             return new FastForestRegression(env, options);
         }
 
@@ -325,12 +334,14 @@ public static FastForestClassification FastForest(this BinaryClassificationConte
             {
                 LabelColumn = labelColumn,
                 FeatureColumn = featureColumn,
-                WeightColumn = weights,
                 NumLeaves = numLeaves,
                 NumTrees = numTrees,
                 MinDocumentsInLeafs = minDatapointsInLeaves,
             };
 
+            if (weights != null)
+                options.WeightColumn = Optional<string>.Explicit(weights);
+
             return new FastForestClassification(env, options);
         }