-
Notifications
You must be signed in to change notification settings - Fork 28.9k
[SPARK-14409][ML] Adding a RankingEvaluator to ML #12461
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
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.
It would be good to expand the doc string here with more detail about what is MRR and how it is computed (see the ndcg doc string above).
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.
Thanks @MLnick I just updated the docs.
|
ok to test |
|
Test build #56105 has finished for PR 12461 at commit
|
|
@MLnick I added the implementation for ml.evaluation.RankingEvaluator. Please let me know if there are any issues. |
|
Test build #56317 has finished for PR 12461 at commit
|
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.
these columns must in fact be ArrayType
|
Test build #56952 has finished for PR 12461 at commit
|
|
Test build #57506 has finished for PR 12461 at commit
|
Until apache/spark#12461 is merged, the ML pipeline doesn’t have the same ability to generate top recommendations for a set of users. So, we’ll tune the model with an ML pipeline, build it from the same data with the RDD-based ALS implementation, then have a component to load that model and generate recs into an intermediate store TBD.
This patch tries to add the implementation of Mean Rreciprocal Rank (MRR) in mllib.evaluation, as a first step toward adding a RankingEvaluator to ML. Additional test cast has been added to cover Mean Rreciprocal Rank (MRR). This patch is a work in progress.
This patch tries to add the implementation of RankingEvaluator in ml.evaluation. Additional test cast has been added to cover the implementation.
This patch tries to update the comment for mean reciprocal rank.
This patch tries to consolidate ml.evaluation and mllib.evaluation so that RankingEvaluator wraps RankingMetrics.
|
Test build #73804 has finished for PR 12461 at commit
|
|
/cc @daniloascione please take a look. |
|
Can one of the admins verify this patch? |
|
Hi, is there any plans for merging this implementation of RankingEvaluator?. I would be glad to help if needed. Thanks. |
|
WYSIWYG here. This one was abandoned and I don't see any other activity. If you want to take a crack at a clean change that follows on this, sure, we can look at it. |
What changes were proposed in this pull request?
This patch adds the implementation of Mean Rreciprocal Rank (MRR) in mllib.evaluation, and adds a RankingEvaluator to ML (ml.evaluation)
How was this patch tested?
Additional test cast has been added to cover Mean Rreciprocal Rank (MRR).