-
Notifications
You must be signed in to change notification settings - Fork 1.9k
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
Enabling custom groupId column in the Ranking AutoML experiment #5246
Conversation
Codecov Report
@@ Coverage Diff @@
## master #5246 +/- ##
==========================================
+ Coverage 73.51% 73.55% +0.04%
==========================================
Files 1013 1022 +9
Lines 188446 189693 +1247
Branches 20289 20441 +152
==========================================
+ Hits 138528 139534 +1006
- Misses 44426 44619 +193
- Partials 5492 5540 +48
|
src/Microsoft.ML.AutoML/Experiment/MetricsAgents/BinaryMetricsAgent.cs
Outdated
Show resolved
Hide resolved
src/Microsoft.ML.AutoML/Experiment/MetricsAgents/IMetricsAgent.cs
Outdated
Show resolved
Hide resolved
@@ -57,9 +57,9 @@ public bool IsModelPerfect(double score) | |||
} | |||
} | |||
|
|||
public RankingMetrics EvaluateMetrics(IDataView data, string labelColumn) | |||
public RankingMetrics EvaluateMetrics(IDataView data, string labelColumn, string groupIdColumn) |
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.
Can we just put groupId column as an option when creating RankingMetricsAgent? So that we don't have to add an extra groupIdColumn to IAgent.
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.
Yes, we can do that. I think a possible concern is that GroupId is required by RankingTrainers, and passing it in as an option gives me the impression that it's optional. I think that if we make the default value of GroupId
"GroupId" and document that, it might be okay. What you do think?
No description provided.