-
Notifications
You must be signed in to change notification settings - Fork 250
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
Add mean row aggregation to HELM summarize #2997
Conversation
@@ -119,6 +119,9 @@ class MetricGroup(Field): | |||
hide_win_rates: Optional[bool] = None | |||
"""If set to true, do not compute win rates.""" | |||
|
|||
aggregation_strategies: Optional[List[str]] = None |
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.
@percyliang could you review this change to the schema?
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.
Awesome!
if strategy == "win_rate": | ||
WIN_RATE_AGGREGATION = "mean" | ||
win_rates = compute_aggregate_row_win_rates(table, aggregation=WIN_RATE_AGGREGATION) | ||
description = "How many models this model outperform on average (over columns)." |
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.
nit: outperform -> outperforms (fix pre-existing typo)
Let's merge this first and have @percyliang review post-merge. |
Adds the option for users to do mean metric aggregation as opposed to mean win rate within helm summarize.
In addition it allows for mutliple types of aggregation in parallel (as opposed to just one).