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);