-
Notifications
You must be signed in to change notification settings - Fork 1.9k
Clarify documentation for GetFeatureWeights in TreeEnsembleModelParameters #3339
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
/// <param name="weights">a <see cref="VBuffer{T}"/> where feature weights would be assigned to. | ||
/// The i-th element in <paramref name="weights"/> stores the weight of the i-th feature.</param> | ||
/// <param name="weights">A <see cref="VBuffer{T}"/> to hold the cumulative split gain value for each feature. | ||
/// The i-th element in <paramref name="weights"/> stores the cumulative split gain of the i-th feature.</param> |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thank you @rogancarr! As discussed "weights" is a bit of an unfortunate word for "cumulative split gains," but it's probably too late to change the API at this point.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thank you @rogancarr !
Codecov Report
@@ Coverage Diff @@
## master #3339 +/- ##
==========================================
- Coverage 72.66% 72.66% -0.01%
==========================================
Files 807 807
Lines 145191 145191
Branches 16223 16223
==========================================
- Hits 105503 105502 -1
Misses 35268 35268
- Partials 4420 4421 +1
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
…meters (dotnet#3339) (cherry picked from commit 0bb8163)
This PR clarifies the documentation for
GetFeatureWeights()
inTreeEnsembleModelParameters
. SinceIHaveFeatureWeights
has becomeinternal
, theTreeEnsembleModelParameters
are now the only place where this method is visible, and it is unclear from the API what this returns. This PR adds documentation to specify that the "weights" returned are the cumulative split gains for all the trees in the ensemble.Fixes #1850