diff --git a/src/Microsoft.ML.FastTree/FastTree.cs b/src/Microsoft.ML.FastTree/FastTree.cs index 253c899d5f..ace3298dd8 100644 --- a/src/Microsoft.ML.FastTree/FastTree.cs +++ b/src/Microsoft.ML.FastTree/FastTree.cs @@ -3229,10 +3229,10 @@ private void ToCSharp(InternalRegressionTree tree, TextWriter writer, int node, } /// - /// Copy the weights of all training features to . + /// Get the cumulative split gains for each feature across all trees. /// - /// a where feature weights would be assigned to. - /// The i-th element in stores the weight of the i-th feature. + /// A to hold the cumulative split gain value for each feature. + /// The i-th element in stores the cumulative split gain of the i-th feature. public void GetFeatureWeights(ref VBuffer weights) { var numFeatures = Math.Max(NumFeatures, MaxSplitFeatIdx + 1);